我想在没有任何输入的情况下实现Keras自定义层,只是可训练的权重。
这是到目前为止的代码:
class Simple(Layer):
def __init__(self, output_dim, **kwargs):
self.output_dim = output_dim
super(Simple, self).__init__(**kwargs)
def build(self):
self.kernel = self.add_weight(name='kernel', shape=self.output_dim, initializer='uniform', trainable=True)
super(Simple, self).build()
def call(self):
return self.kernel
def compute_output_shape(self):
return self.output_dim
X = Simple((1, 784))()
我收到一条错误消息:
__call__() missing 1 required positional argument: 'inputs'
在Keras中没有输入的情况下是否可以构建自定义图层的解决方法?