mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-12 16:18:22 +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:
|
def test_none_end_slicing() -> None:
|
||||||
x = torch.randn(2, 1000, 400)
|
x = torch.randn(2, 1000, 400)
|
||||||
slicing = Slicing(dim=1, start=1)
|
slicing_layer = Slicing(dim=1, start=1)
|
||||||
sliced = slicing(x)
|
sliced = slicing_layer(x)
|
||||||
expected = x[:, 1:, :]
|
expected = x[:, 1:, :]
|
||||||
assert torch.equal(sliced, expected)
|
assert torch.equal(sliced, expected)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue