a = torch.tensor([1,5,62,54])
print(a.size())
a= a.view(1,len(a))
print(a)
print(a.size())
predict=torch.max(a,1)[1].numpy()
print(predict)
运行结果:
E:\anaconda\python.exe D:/pycharm_prog/d0127/learning_python.py
torch.Size([4])
tensor([[ 1, 5, 62, 54]])
torch.Size([1, 4])
[2]
the end of learning python