def test_sequence_bias(self):
    x = torch.randn(4, 3, 2)
    layer = SequenceBias(2)
    self._check_one_layer(layer, x, batch_first=False)
def test_sequence_bias(self):
    x = torch.randn(4, 3, 2)
    layer = SequenceBias(2)
    self._check_one_layer(layer, x, batch_first=False)