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

直接上例子

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

正常输出
3.对u进行转置:print(torch.transpose(u))
报错
4.对于以上报错原因分析:pytorch中的transpose()函数需指定转置的维度,即我们这里需要指定0(表示行)和1(表示列)进行转换
5.修改代码:print(torch.transpose(u,0,1))
转置成功
6.成功运行!

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

你可能感兴趣的:(深度学习,python相关,pytorch,深度学习,python)