So what I’m actually trying to do is to get the mean of the list of 6 tensors from different augmented images to be used during testing. (test time augmentation)
My expected shape would be the mean of all 6 tensors, with the same shape
[12 x 7 x 368 x 640]