(self, input_shape)
| 61 | self.dense_layer = [MyDense(i, activation="relu") for i in hidden_units] |
| 62 | |
| 63 | def build(self, input_shape): |
| 64 | self.output_layer = MyDense(input_shape[-1], activation=None) |
| 65 | |
| 66 | def call(self, inputs, **kwargs): |
| 67 | if K.ndim(inputs) != 2: |