How to calculate tensor mean?

1

image

I want to calculate mean by y-axis, so the output should be 1x3.
How should I do?

Thank you.

2 
import torch
x = torch.rand(3, 3)
print(x.mean(1))
1 Like
3

x.mean(0) was what I wanted.
Thank you so much^^!!