Скажем, у меня есть модель CNN, которая выводит N карт вероятностей в виде маски того же размера входного изображения в стиле Unet. Затем я хотел бы применить, например, метод наименьших квадратов сверху каждой маски, чтобы вместо этого получить коэффициенты для функций в качестве выходных данных, и использовать их для расчета потерь в моих моделях.
def unet_model(...)
# init unet model
...
...
# final layer
mask_out = layers.Conv2D(output_channels, (1,1), activation='softmax')(conv9)
# start applying e.g least squares fit here
eq_list = tf.Variable((x_map, y_map, mask_out))
transp = tf.transpose(a)
...
Транспорт получит следующую ошибку, когда я инициализировать модель. Я проверял операции подгонки наименьших квадратов в других местах.
FailedPreconditionError: Error while reading resource variable _AnonymousVar1423 from Container: localhost. This could mean that the variable was uninitialized. Not found: Resource localhost/_AnonymousVar1423/N10tensorflow3VarE does not exist. name: transpose/
У меня есть некоторые предположения, например, что транспонирование не может иметь дело с осью заполнителей для размеров партии, но, как правило, не в курсе этого.