I have a single 2D Kernel of size [3,3], and a Tensor of size [B, 64, H, W]
How can I apply this 2D kernel on each channel?
How should i reshape/repeat the kernel?
I tried to repeat my Kernel, as follows:
kernel = kernel.repeat((B, 64, 1, 1))
But when I apply it the tensor size changes
[1, 64, H, 1]
I am forced to do this for now but it is inefficient:
    for b in range(0, dpv_permuted.shape[0]):
        for c in range(0, dpv_permuted.shape[1]):
            dpv_permuted[b,c,:,:] = F.conv2d(dpv_permuted[b,c].unsqueeze(0).unsqueeze(0), **spread_kernel).squeeze(0).squeeze(0)