适用于单个样本的图层到适用于批处理的图层

Question

我目前有一个如下所示的图层：class MyLayer(tf.keras.layers.Layer):    def __init__(self):        super(MyLayer, self).__init__()    def call(self, img, params):        tf.foo(img)        tf.bar(img, params)        return imgcall 方法img使用 shape(128, 128, 3)和paramsshape获取和输入(15)。我必须更改什么才能使图层可以批量操作？img例如，输入将具有 shape(32, 128, 128, 3)并且params将具有 shape (32, 15)。所以问题基本上是：我必须如何编辑图层，使其执行与现在相同的操作，但对于批处理中的每个图像？

Answer 1

也许你只是想重塑图像张量的大小？你想要做的就是 x=np.reshape(x,(32,128,128,3))。让我知道这是否有帮助。