Answers for "torch mean of tensor"

1

mean of torch tensor

torch.mean(input)
Posted by: Guest on February-23-2021
0

torch mean of tensor

>>> a = torch.randn(1, 3)
>>> a
tensor([[ 0.2294, -0.5481,  1.3288]])
>>> torch.mean(a)
tensor(0.3367)
Posted by: Guest on April-26-2022

Python Answers by Framework

Browse Popular Code Answers by Language