torch.item()的用法

理解为把tensor量变成常数每次从矩阵中取一个

import torch
x = torch.arange(1,4).view(2,2)
print(x[1,1])
print(x[1,1].item())

输出

tensor(4.)
4.0