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)
Torch的Tensor与numpy的array会共享储存空间,修改一个也会导致另外一个被修改
a = torch.ones(5)
b = a.numpy()
a.add_(1)
print(a)
print(b)
a = np.ones(5)
b = torch.from_numpy(a)
np.add(a,1,out=a)
print(a)
print(b)