torch.item()和torch.data的区别

当tensor中只有一个值的时候 .item()返回一个数值,.data返回一个tensor

使用例:

  test = torch.Tensor(42)
  a = test.item()
  b = test.data
  print(f'a:{a}, b:{b}')
  #输出为:a:42, b:tensor(42)

torch 低版本的单值tensor不会自动转换为数值类型,
为保证兼容性,求和后尽量使用.item

posted @ 2021-09-29 09:33  小艾衰  阅读(821)  评论(0编辑  收藏  举报