如何在 Keras 中使用 fft 创建自定义损失函数答案

【问题标题】：How to create custom loss function with fft in Keras如何在 Keras 中使用 fft 创建自定义损失函数
【发布时间】：2020-02-20 11:55:52
【问题描述】：

我想计算一个损失函数，但我在使用 fft 函数时遇到了问题。这里是 numpy 的代码：

def bidule(y_true, y_pred):
product = np.fft.fft(y_true) * np.fft.fft( y_pred).conj()
cc = np.fft.fftshift(np.fft.ifft(product))
return np.argmax(cc.real)

我想使用 Keras 后端，但是当我将 np 替换为 K 时，代码无法运行。 “模块‘keras.backend’没有属性‘fft’”

有人可以帮帮我吗？

谢谢

尼古拉斯

【问题讨论】：

标签： tensorflow keras fft recurrent-neural-network loss-function

【解决方案1】：

请先检查Keras Backend Documentation 看是否有.fft 属性。 keras.backend 中没有 .fft。但是，您可以通过使用带有 Keras 的 Lambda 层来使用 TensorFlow 实现带有 tf.signal.fft 的一个。在以下link 中使用Lambda 和光谱傅立叶函数的一个示例。您可以使用给定的链接实现您的。

【讨论】：