mean(): could not infer output dtype. Input dtype must be either a floating point or complex dtype.

一、参考资料

1.数据类型出错

2.数据类型转化

二、问题描述

        在进行torch学习的时候，遇见了不同创建的张量，最后一个能使用mean()函数，另一个不能使用mean()函数。

m=torch.arange(15).reshape(3,5)
n=torch.randn((3,5))
m,n
m.mean()
n.mean()

三、出错原因

input：输入张量。它的数据类型必须是浮点型或复数型。对于复数的输入，范数使用每个元素的绝对值。注意，输入张量中元素的数据类型一定得是浮点型或者是复数。这就是报错原因。

四、解决方案

1.将张量的数据类型修改type(torch.type):

m=torch.arange(15).reshape(3,5)

n=torch.randn((3,5))

l=m.type(torch.float64)

l.mean()

成功解决。

ps:long()方法不行。