Inspired by this 33 answer on SO, I tried converting my code to use the conv3d functional but it gives me a weird error.
This is the code:
z = F.conv3d(y.unsqueeze(0), x )
The sizes are:

x: torch.Size([4, 16, 6, 5, 5])
y: torch.Size([4, 6, 14, 14])

The error I get is
RuntimeError: CUDNN_STATUS_BAD_PARAM
Thanks