.data.max用于找概率最大的下标
c = torch.tensor([[1,2,3],[4,5,6]])
print(c)
d = c.data.max(1,keepdim=True)[1]
print(d)
输出:
tensor([[1, 2, 3],
[4, 5, 6]])
tensor
.data.max用于找概率最大的下标
c = torch.tensor([[1,2,3],[4,5,6]])
print(c)
d = c.data.max(1,keepdim=True)[1]
print(d)
输出:
tensor([[1, 2, 3],
[4, 5, 6]])
tensor