Я хочу реализовать пользовательский слой 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?