Thanks for the addition, @KFrank!
The segfault might be related to this issue 7 and should be fixed per this PR 4.



 yegane:

e.g. if I have ten models, that I want to aggregate their models parameters, I could simply multiply them by 0.1 and then use torch.sum, and I expect to get the same result as torch.mean; yet I am not getting the same results. It is my simplest scenario, but I have the same problem in more complex operations that I want to compare.


You could try to force a specific order via loops, which might be slower, but would probably give you exactly the same results. Also, float64 might be another good idea and I would try @KFrank’s suggestion on transforming the tensors.