0

私の PyTorch プログラムには、実行時に継続的に更新されるマトリックスがあります。

この更新をどのように実行するのだろうか。私はこのようなものを使用してみました:

matrix[0, index] = hidden[0]

matrixとはどちらhiddenも autograd 変数です。上記の例を使用すると、次のエラー メッセージが表示されます。

RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation

これを回避して、インプレース操作を使用せずに更新を実行する方法を知りたいです。

おそらく、マトリックスをスライスして torch.cat で新しいマトリックスを作成するとうまくいくでしょうが、これはあまり良い解決策ではないようです。

これを行うより良い方法はありますか?

前もって感謝します!

4

1 に答える 1