Torch的Tensor与numpy的array数据相互转换

        Torch的Tensor与numpy的array会共享储存空间,修改一个也会导致另外一个被修改

Tensor转换成Numpy数组

a = torch.ones(5)
b = a.numpy()

a.add_(1)

print(a)
print(b)

将Numpy数组转换成Tensor

a = np.ones(5)
b = torch.from_numpy(a)
np.add(a,1,out=a)

print(a)
print(b)
发布了54 篇原创文章 · 获赞 80 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_41685265/article/details/104713578
今日推荐