(self, x)
| 197 | return x |
| 198 | |
| 199 | def num_flat_features(self, x): |
| 200 | size = x.size()[1:] # all dimensions except the batch dimension |
| 201 | num_features = 1 |
| 202 | for s in size: |
| 203 | num_features *= s |
| 204 | return num_features |
| 205 | |
| 206 | |
| 207 | ############################################################################ |