如何将numpy数组转换为Torch张量

将numpy数组转换为Torch张量,看到如何更改np阵列自动更改
import numpy as np
a = np.ones(5)
b = torch.from_numpy(a)
np.add(a, 1, out=a)
print(a)
print(b)
你可以中文网文档:https://ptorch.com/docs/3/tens ... ensor
已邀请:

要回复问题请先登录注册