Ok, i try to explain myself correctly.
In the image below, i would like to extract a tensor from each N plane (red points) starting from the first column/rown and so on. So, in this way i have to obtain X tensors composed by each planes (conv2). Is it possible with pytorch?

LeNet-5-19981552×595 41.2 KB