_th_addr_out CPUType 不支持 ComplexFloat

Question

我正在尝试为我的神经网络使用自定义的损失函数。我已经在 torch 中实现了所有操作，并且我的数据中有复数。

我在训练 NN 时遇到错误：

RuntimeError: _th_addr_out not supported on CPUType for ComplexFloat 

你知道任何可能的解决方案吗？

Answer 1

好吧，PyTorch 中的 Complex Autograd 目前似乎处于原型状态，并且不包括某些功能的向后功能。

例如：torch.sign，用于torch.abs的反向计算，没有为复数张量定义。 torch.mv 也一样。所以我逐行调试我的代码，找出没有包含的函数，并用自定义函数替换它们:)

希望 PyTorch 的下一个版本中包含更多功能。