欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

TypeError: transpose() received an invalid combination of arguments - got (Tensor)的问题

程序员文章站 2022-05-27 12:16:53
...

直接上例子

1.u=torch.FloatTensor([[1,0],[0.5,0.2],[0,1]])
2.print(u)

TypeError: transpose() received an invalid combination of arguments - got (Tensor)的问题
3.对u进行转置:print(torch.transpose(u))
TypeError: transpose() received an invalid combination of arguments - got (Tensor)的问题
4.对于以上报错原因分析:pytorch中的transpose()函数需指定转置的维度,即我们这里需要指定0(表示行)和1(表示列)进行转换
5.修改代码:print(torch.transpose(u,0,1))
TypeError: transpose() received an invalid combination of arguments - got (Tensor)的问题
6.成功运行!

虽然是很简单的问题,但因为自己第一次在这块搞了好久,谨此记录!(最重要的是还是说明自己基础太差了^ ~ ^)