Solved by AlanSouthwark in post #2 
              
              
                Anyway, I didn’t understand the argument “dim” correctly, with “dim” argument “torch.mean(T, dim=i)” do everything for me, but for getting the mean of each column, i=0 is the correct dimension.