Thanks for this quick response. You’re right, it’s not in the binary release. I will try compile with the source. Also, it doesn’t do what I want. For instance, I have a 1x3x4x4 tensor, what I would like to do is to compute the mean of the 3 channels so that I get a tensor of size 1x3x1x1, while torch.mean will only calculate mean of a single dimension not cross two dimensions.
>>> import torch
>>> a = torch.randn(1, 3, 4, 4)
>>> torch.mean(a)
-0.1468020509540414
>>> torch.mean(a,dim=1)
(0 ,0 ,.,.) = 
 -0.3284  0.1365 -0.4091 -0.6045
 -0.3535 -0.8380  0.0724  0.4533
 -0.2629  0.7440  0.2587 -0.0283
 -0.5283 -0.3177 -0.4128  0.0697
[torch.FloatTensor of size 1x1x4x4]

>>>