(self, X_shape)
| 157 | self.input_dim = None |
| 158 | |
| 159 | def setup(self, X_shape): |
| 160 | self.dense = Dense(self.output_dim) |
| 161 | self.dense.setup((X_shape[0], X_shape[2])) |
| 162 | self.input_dim = X_shape[2] |
| 163 | |
| 164 | def forward_pass(self, X): |
| 165 | n_timesteps = X.shape[1] |