Fatal编程技术网
  • pytorch/
  • Pytorch _ComplexFloat的CPUType不支持th_addr_out

Pytorch _ComplexFloat的CPUType不支持th_addr_out

pytorch

Pytorch _ComplexFloat的CPUType不支持th_addr_out,pytorch,customization,complex-numbers,loss-function,autograd,Pytorch,Customization,Complex Numbers,Loss Function,Autograd,我正在尝试为我的NN使用自定义损失函数。我已经在torch中执行了所有操作，我的数据中有复杂的数字 我在训练NN时出错： RuntimeError: _th_addr_out not supported on CPUType for ComplexFloat 您知道任何可能的解决方案吗？好吧，PyTorch中的复杂Autograd目前似乎处于原型状态，一些功能的向后功能不包括在内 例如：torch.abs的反向计算中使用的torch.sign没有为复数张量定义。torch.mv也一样。因此，

我正在尝试为我的NN使用自定义损失函数。我已经在torch中执行了所有操作，我的数据中有复杂的数字

我在训练NN时出错：

RuntimeError: _th_addr_out not supported on CPUType for ComplexFloat
您知道任何可能的解决方案吗？

好吧，PyTorch中的复杂Autograd目前似乎处于原型状态，一些功能的向后功能不包括在内

例如：torch.abs的反向计算中使用的torch.sign没有为复数张量定义。torch.mv也一样。因此，我逐行调试代码以查找未包含的函数，并用自定义函数替换它们：）

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