mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
Update tests/fluxion/layers/test_basics.py
Co-authored-by: Cédric Deltheil <355031+deltheil@users.noreply.github.com>
This commit is contained in:
parent
7d9ceae274
commit
e2f2e33add
|
@ -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)
|
||||
|
||||
|
|
Loading…
Reference in a new issue