Ошибка построения слоя Keras: build () принимает 1 позиционный аргумент, но два

Question

У меня есть следующая ошибка в этом простом слое:

class MyLayer(Layer):

def __init__(self):
    super(MyLayer, self).__init__()

def build(self):
    # Create a trainable weight variable for this layer.
    self.kernel = self.add_weight(name='kernel', 
                                  shape=(1)
                                  trainable=True)
    super(MyLayer, self).build() 

def call(self, x):
    return x/self.kernel

Когда я использую его как:

m = MyLayer()
t = m (input)

Ошибка: build () принимает один позиционный аргумент, нодва были даны.

Answer 1

Каждый слой в слое Keras требует аргумент input_shape. Добавьте его к вашему build() методу.