(self, units, activation=None, **kwargs)
| 6 | |
| 7 | class MyDense(tf.keras.layers.Layer): |
| 8 | def __init__(self, units, activation=None, **kwargs): |
| 9 | super(MyDense, self).__init__(**kwargs) |
| 10 | self.units = units |
| 11 | self.w = None |
| 12 | self.activation = activation |
| 13 | |
| 14 | def build(self, input_shape): |
| 15 | self.w = self.add_weight( |