Я написал функцию с использованием тензорного потока, например, y = f (x), и она хорошо работает с одним изображением, например (1, 4, 4, 1). Теперь я не знаю, как применить это к изображению или тензору типа x = (1, 4, 4, 32). Если бы это был Python или C ++, я мог бы использовать цикл for с 32 итерациями и применить свою функцию к каждому срезу и объединить их в одно изображение в виде, подобном y = (1, 4, 4, 32).
Для лучшего объяснения, представьте
y = tf.keras.layers.Conv2D(filters=32, kernel_size=(3, 3), activation='relu', padding='same')(x)
как это работает на фильтрах? Это применяется один раз для каждого фильтра или нет? Есть ли в тензорном потоке функция, которая применяет один оператор на каждую глубину и объединяет их?
Я думал использовать while_loop, но это сложно в тензорном потоке. Если это единственный способ, может кто-нибудь дать мне знать, как работать с циклом while на простом примере с использованием x, f и y?
Любая помощь будет оценена.