I have no idea how torch.mean(dim=0) is working.
for example when i run the code below
#------------------------------------------------
import torch
t = torch.FloatTensor([[1, 2], [3, 4]])
print(t.mean(dim=0)
#------------------------------------------------
The output is
tensor([2., 3.])
I don’t know why the output comes like this.
Anyone with the insight please tell me how it is working like this.