如何在不使用就地操作的情况下交换 PyTorch 张量中的值（保留梯度）

Question

我有一个名为state的张量，形状为torch.Size([N, 2**n, 2**n]) ，我想应用以下操作：

state[[0,1]] = state[[1,0]]
state[0] = -1*state[0]

这两个都是就地操作。 是否有一些不合适的操作可以替代它们？ 这些行位于 for-loop 内，因此仅创建新变量会有点困难。

Answer 1

我设法弄明白了！

代替：

state[[0,1]] = state[[1,0]] # in-place operation

和：

state = state[[1,0]] # out-of-place operation

对于第二行，我们替换：

state[0] = -1*state[0] # in-place operation

和：

# out-of-place operations
temp = torch.ones(state.shape).type(state.type()).to(state.device)
temp[1] = -1*temp[1]
state = state*temp

这似乎在做这项工作！