def test_lstm_batch_first(self):
    layer = DPLSTM(25, 12, 1, batch_first=True)
    x = torch.randn(30, 20, 25)
    self._check_one_layer(layer, x, batch_first=True)
def test_lstm_batch_first(self):
    layer = DPLSTM(25, 12, 1, batch_first=True)
    x = torch.randn(30, 20, 25)
    self._check_one_layer(layer, x, batch_first=True)