(self, X_train)
| 914 | self.is_initialized = True |
| 915 | |
| 916 | def forward(self, X_train): |
| 917 | Xs = {} |
| 918 | out = X_train |
| 919 | for k, v in self.layers.items(): |
| 920 | Xs[k] = out |
| 921 | out = v.forward(out) |
| 922 | return out, Xs |
| 923 | |
| 924 | def backward(self, grad): |
| 925 | dXs = {} |