当用jupyter notebook写代码的时候遇到了上述错误,表示的意思是函数mean()只能作用于浮点型或者复数,这时候检查代码:
A = torch.arange(20*2).reshape(2,5,4)
可以发现我的张量A是整型,因此我们将其改写为浮点型,代码如下:
A = torch.arange(20*2,dtype=torch.float32).reshape(2,5,4)
问题解决!
当用jupyter notebook写代码的时候遇到了上述错误,表示的意思是函数mean()只能作用于浮点型或者复数,这时候检查代码:
A = torch.arange(20*2).reshape(2,5,4)
可以发现我的张量A是整型,因此我们将其改写为浮点型,代码如下:
A = torch.arange(20*2,dtype=torch.float32).reshape(2,5,4)
问题解决!