Solved by albanD in post #2 
              
              
                Hi, 
I’m not sure to understand what you’re trying to do here. You can get the same result as your zip_and_stack_tensors() using a single torch.stack(). Am I missing something? 
import torch
from torch import nn, optim

def zip_and_stack_tensors(tensor1, tensor2):
    num_of_channels = len(tensor1[0…