Torch优化包可以支持多个输入

Question

我正在尝试使用火炬7 optim package adam algorithm implementation 用于优化需要两个独立输入的神经网络。可以这样做吗？代码似乎只支持单个输入向量。是否有一些其他实现可以采用通用的输入表？ 我看到的参考用法是here

Answer 1

原则上，如果将所有参数和渐变展平为两个张量，则optim可以处理它。请参阅getParameters()功能。