我使用tensorflow编写了一个函数,例如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,但是在tensorflow中很困难。如果这是唯一的方法,有人可以通过使用x,f和y的简单示例让我知道如何使用while循环吗? 任何帮助,将不胜感激。