(self, grad)
| 1136 | return out, Xs |
| 1137 | |
| 1138 | def backward(self, grad): |
| 1139 | dXs = {} |
| 1140 | out = grad |
| 1141 | for k, v in reversed(list(self.layers.items())): |
| 1142 | dXs[k] = out |
| 1143 | out = v.backward(out) |
| 1144 | return out, dXs |
| 1145 | |
| 1146 | def update(self): |
| 1147 | """ |