在torch.tensor中使用for循环时，如何更改特定值？

Question

在实现本地差异隐私时，我对这段代码的逻辑感到困惑。

我正在尝试使用以下给定的逻辑来实现：

当第一个硬币翻转为0时，输出应与第二个硬币翻转完全相同。

当第一个硬币翻转为1时，输出应与条目本身完全相同。

n

用于硬币翻转； 0：尾巴和1：头部

在torch.tensor中使用for循环时，如何更改特定值？ （例如：对于db中的条目，检查一条语句，然后更改条目）

我已经通过 torch.tensor文档尝试了许多类似coin flips: 12 entries: 0 1 00 0 | 0 01 1 | 1 10 0 | 1 11 0 | 1 的操作。但是，到目前为止无法解决。

torch.tensor.where

预期输出应该像我上面提到的那样。但是实际输出是我不想要的，因为我没有更改 db 张量中的特定 entry 值。