Torch.mean() in two dimensions

1

a tensor with size (a,b,c,d) how to average and reduce it into size (a) using torch.mean()

2

How about this?

tensor.view(a, -1).mean(1)
3

Does starting by view first keep the trace of the gradient ?

4

If you use it on a Variable, then yes.

1 Like