torch.item()的用法
torch数据类型的item方法是得到只有一个元素张量里面的元素值。如下:>>> x = torch.tensor(4)>>> x.item()4如果对包含多个元素的torch.tensor用item()方法,则会报错如下:>>> x = torch.tensor([1,2,3,4])>>> x.item()Traceback (most recent call last): File "<stdin>
原创
2021-03-25 18:10:52 ·
5389 阅读 ·
3 评论