def build(self, input_shape):
    self.kernel = self.add_weight(name='kernel', shape=[self.output_dim, input_shape[(- 1)]], initializer='zeros', trainable=True)
    self.bias = self.add_weight(name='bias', shape=[self.output_dim], initializer='zeros', trainable=True)
def build(self, input_shape):
    self.kernel = self.add_weight(name='kernel', shape=[self.output_dim, input_shape[(- 1)]], initializer='zeros', trainable=True)
    self.bias = self.add_weight(name='bias', shape=[self.output_dim], initializer='zeros', trainable=True)