如何在不使用就地操作的情況下交換 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

這似乎在做這項工作！