Pytorch的item()方法

pytorch中.item()方法 是得到一个元素张量里面的元素值

import torch
torch.arange(3)

tensor([0, 1, 2])

 

torch.arange(3)[0].item()

0

注：返回元素0.