I have two tensors, A and B, with sizes
A.size()
>> torch.Size([10, 5, 5, 5])

B.size()
>> torch.Size([4])

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!
Best regards