如何解决如果割炬功能未包含其某些操作,如何在Pytorch中实现损失功能?
如果Pytorch的某些操作未包含在Torch函数中,如何在Pytorch中实现损失函数?
例如我有一个名为Net的网络和一个名为Input的输入,然后我们可以计算出= Net(Input)。
通常,我们可以通过loss = loss_fn(out,ground_truth)计算损失。
但是,如果我需要在out
上做什么?就像,我想首先对其应用傅里叶变换。由于pytorch函数不能很好地支持FFT,因此我可能需要更改为numpy才能执行此操作。我尝试了这个,但是向后的损失似乎很奇怪。谁能告诉我正确的方法?非常感谢:)
解决方法
暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!
如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。
小编邮箱:dio#foxmail.com(将#修改为@)