Solved by cll in post #3 
              
              
                Sure! Please see the code below. The 2D Convolution block appears to work well. I have since managed to implement the 3D Convolution block using a similar approach, however, it only appears to function when padding=(0, 0, 0). I’m using torch.Tensor.unfold to unfold 5D input tensors. Unfortunately, i…