I have two tensors, B and a predefined kernel.
B.size()
>> torch.Size([6, 6, 1])

kernel.size()
>> torch.Size([5])

In scipy it’s possible to convolve the tensor with the kernel along a single axis like:
convolve1d(B.numpy(),kernel.numpy(), axis=0,mode="constant")
mode="constant" refers to zero-padding.
torch.nn.functional.conv1d, however, doesn’t have a parameter to convolve along a single axis. Is it possible to mimic that behaviour of scipy?
Thank you in advance!