Как создать собственную функцию потерь с помощью fft в Keras

Question

Я бы хотел вычислить функцию потерь, но у меня возникла проблема с использованием функции 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 ""

Может кто-нибудь помочь мне, пожалуйста?

Спасибо

Николас

Answer 1

Сначала проверьте документацию Keras Backend , чтобы узнать, есть ли атрибут .fft. В keras.backend нет .fft. Однако вы можете реализовать его с tf.signal.fft, используя TensorFlow, используя слой Lambda с Keras. Один пример использования Lambda со спектральной функцией Фурье в следующей ссылке . Вы можете реализовать свой по данной ссылке.