刺无名客的博客

最初学的时候有点潦草，不多说，进例子 首先来个二维的 import torch a=torch.tensor([[1,2,3],[-1,3,2]]) print(a.shape) b=torch.argmax(a,dim=0) print(b) ''' torch.Size([2, 3]) tensor([0, 1, 0]) ''' import torch a=to...