torch.tensor vs torch.from_numpy()

发布时间:2023年12月18日

torch.tensor函数和torch.from_numpy()函数都可以将numpy array转换为torch tensro,不同之处在于前者共享内存(torch的底层实现就是numpy,所以可以共享),后者开辟新的空间

a=np.ones(4)
b=torch.from_numpy(a)

a[0]=2
b[1]=2
print(a,b)

output:
[2. 2. 1. 1.] tensor([2., 2., 1., 1.], dtype=torch.float64)

?
d=np.ones((4))
e=torch.tensor(d)
print(d,e)

d[0]=2
print(d,e)
e[1]=2
print(d,e)

output:
[1. 1. 1. 1.] tensor([1., 1., 1., 1.], dtype=torch.float64)
[2. 1. 1. 1.] tensor([1., 1., 1., 1.], dtype=torch.float64)
[2. 1. 1. 1.] tensor([1., 2., 1., 1.], dtype=torch.float64)

?
文章来源:https://blog.csdn.net/m0_51312071/article/details/135073539
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。