Torch 优化包是否支持多个输入

Question

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

Answer 1

原则上，如果您将所有参数和梯度展平为两个张量，那么 optim 可以处理它。请参阅 getParameters() 函数。