I have two tensors,
B, with sizes
A.size() >> torch.Size([10, 5, 5, 5]) B.size() >> torch.Size()
In scipy it’s possible to perform a convolution of the two along a specific axis by
scipy.ndimage.convolve1d(A, B, axis=0)
Is this possible with pytorch? I’ve been trying to make it work with
torch.nn.functional.conv1d, but haven’t succeded.
Thanks a lot for your time!