двусмысленность формы с лямбда-функциями

Question

У меня есть слой tf.keras.layers.Lambda, который использует функции tf.Почему иногда мне приходится указывать output_shape=() для моего Lambda вызова, а иногда нет?

Пример

freqDomain = tf.keras.layers.Lambda(tf.fft2d,
                                    output_shape=(None, tileSize, tileSize, 1))(cpxInput) # Won't work unless output_shape is specified.

Answer 1

Это ошибка в Керасе, в частности около tf.ff2td прямо сейчас.

Keras не может автоматически вычислять форму вывода для функций fft2d в данный момент на основе ввода, поэтому необходимо указать output_shape. Над этим будут работать разработчики:

https://github.com/keras-team/keras/issues/12121#issuecomment-457217337