Update tests/fluxion/layers/test_basics.py

Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
This commit is contained in:
Benjamin Trom 2023-12-13 15:38:35 +01:00
parent 7d9ceae274
commit e2f2e33add

View file

@ -21,8 +21,8 @@ def test_slicing_negative_indices() -> None:
def test_none_end_slicing() -> None:
x = torch.randn(2, 1000, 400)
slicing = Slicing(dim=1, start=1)
sliced = slicing(x)
slicing_layer = Slicing(dim=1, start=1)
sliced = slicing_layer(x)
expected = x[:, 1:, :]
assert torch.equal(sliced, expected)