我目前有一個如下所示的圖層：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)。所以問題基本上是：我必須如何編輯圖層，使其執(zhí)行與現(xiàn)在相同的操作，但對于批處理中的每個圖像？