Keras和Tensorflow：复杂的自定义损失函数的梯度

Question

我已经编写了一个自定义损失函数，该函数计算y_pred和y_true的傅立叶逆变换的MSE。这些y_pred和y_true都是形状的张量（batch_size，256,256,2），其中最后的2个通道是实部和虚部。然后，我使用real + j * imag构建一个复杂的张量，并使用它来计算ifft2。

张量流如何为此计算梯度？网络训练了，但是我不知道它如何为该损失函数计算梯度，以及它是否正确。