torch.tensor vs torch.from_numpy()

2023-12-18 23:29:49

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
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。