Pytorch中的操作是否新结果分配内存的几个问题

自己给自己赋值操作

问题1

before = id(Y)
Y = Y+X
id(Y) = before

输出：False

解决方法

Z = torch.zeros_like(Y)
id_before = id(Z)
Z[:] = X+Y
id_later = id(Z)
id_before = id_later

输出：True

问题2-reshape

a = torch.arange(12)
b = a.reshape((3,4))
b[:] = 2
a

输出：[2,2,2,2,2,2,2,2,2,2,2,2,2,2]
reshape的操作相当于创建了一个指针b

解决方法

b = a.clone()