import torch
a = torch.tensor([1, 2, 3])
print(a) # tensor([1, 2, 3])
print(a.item()) # 会报错,因为 item() 只适配输出单个变量,无法适配list或者np.ndarray
print(a.numpy()) # [1, 2, 3]
【pytorch】tensor
最新推荐文章于 2024-08-30 18:07:10 发布
import torch
a = torch.tensor([1, 2, 3])
print(a) # tensor([1, 2, 3])
print(a.item()) # 会报错,因为 item() 只适配输出单个变量,无法适配list或者np.ndarray
print(a.numpy()) # [1, 2, 3]