qq_38680752的博客

>>> import torch >>> x = torch.arange(15).view(3,5)*1.0 #乘1.0因为torch.mean只能处理float类型 >>> print(x) tensor([[ 0., 1., 2., 3., 4.], [ 5., 6., 7., 8., 9.], [10., 11., 12., 13., 14.]]) torch.mean(input, dim,