Keras:无输入的自定义图层

时间:2019-03-21 14:16:18

标签: python tensorflow keras

我想在没有任何输入的情况下实现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中没有输入的情况下是否可以构建自定义图层的解决方法?

0 个答案:

没有答案