Ignore dimension 0 when calculating mean

1

Is there any way I can calculate the mean of all elements in a 4d tensor except along dim 0 - i.e. the equivalent of tensor.mean(1).mean(1).mean(1)?

2

if you know the size of the first dimension you can:

tensor.view(size,-1).mean(1)

1 Like
3

And if you don’t know it, you could use:

tensor.view(tensor.size(0), -1).mean(1)

:wink:

2 Likes
4

You can also do:

tensor.flatten(1).mean(1)