繁体   English   中英

PyTorch-修改autograd变量

[英]PyTorch - modifications of autograd variables

在我的PyTorch程序中,我有一个在运行时不断更新的矩阵。

我不知道如何执行此更新。 我试过使用这样的东西:

matrix[0, index] = hidden[0]

matrixhidden都是autograd变量。 使用上面的示例时,我收到以下错误消息:

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

我想知道如何解决此问题并在不使用就地操作的情况下执行更新。

可能切片矩阵并使用torch.cat创建一个新的矩阵将起作用,但这似乎不是一个很好的解决方案。

有更好的方法吗?

提前致谢!

也许发布一段代码可能会有所帮助,但是您是否尝试过使用数据集? 您可以使用它来有效地顺序运行数据。

http://pytorch.org/docs/master/data.html#torch.utils.data.TensorDataset

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM