python - Keras 层构建错误 : build() takes 1 Positional Argument but two were given

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)

Error: build() takes one positional argument but two were given.

Answer 1

Keras 层中的每一层都需要一个 input_shape 参数。将它添加到您的 build() 方法中。