PyTorch-修改autograd變量

Question

在我的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創建一個新的矩陣將起作用，但這似乎不是一個很好的解決方案。

有更好的方法嗎？

提前致謝！

Answer 1

也許發布一段代碼可能會有所幫助，但是您是否嘗試過使用數據集？ 您可以使用它來有效地順序運行數據。