pytorch 就地操作 显存节省
X = torch.arange(12, dtype=torch.float32).reshape((3, 4))Y = torch.tensor([[2.0, 1, 4, 3], [1, 2, 3, 4], [4, 3, 2, 1]])示例1before = id(Y)Y = Y + Xid(Y) == before以上代码输出结果为False。这表明进行矩阵加操作前的矩阵Y和进行操作后的矩阵Y本质上存储地址是不一样的。这主要是因为进行Y = Y + X操作时,pytorch会进行.