Source code for norse.torch.module.test.test_conv
import torch
from norse.torch.module.conv import LConv2d
[docs]def test_lconv():
input_tensor = torch.randn(100, 20, 16, 10, 50)
m = LConv2d(16, 33, (3, 5))
assert m(input_tensor).shape == torch.Size([100, 20, 33, 8, 46])