For example, I have 6 tensors with different dimension:
tensor11 size: (b, 64, 224, 224)
tensor21 size: (b, 128, 112, 112)
tensor31 size: (b, 256, 56, 56)
tensor12 size: (b, 64, 224, 224)
tensor22 size: (b, 128, 112, 112)
tensor32 size: (b, 256, 56, 56)
Could I get the tensors which tensor1 has the size (tensor11 size, tensor12 size, tensor13 size), tensor2 has the size(tensor12 size, tensor22 size, tensor 32 size).
If possible, how could I use nn.MSELoss() to computer the loss between elements between these 2 tensor on same positions and get these 3 losses between (tensor11, tensor12), (tensor21, tensor22) and (tensor31, tensor32) by one step while these losses will be utilised in BP?