Я кодировал пользовательскую функцию потерь, которая вычисляет MSE обратного преобразования Фурье для y_pred и y_true. Эти y_pred и y_true являются тензорами формы (batch_size, 256,256,2), где 2 канала в конце являются действительной и мнимой частью. Затем я создаю сложный тензор с вещественным + j * imag и использую его для вычисления ifft2.
Как может тензорный поток вычислить градиенты для этого? Сеть тренируется, но я понятия не имею, как она может вычислить градиенты для этой функции потерь и если она верна.