(self, x)
| 172 | return x |
| 173 | |
| 174 | def num_flat_features(self, x): |
| 175 | size = x.size()[1:] # all dimensions except the batch dimension |
| 176 | num_features = 1 |
| 177 | for s in size: |
| 178 | num_features *= s |
| 179 | return num_features |
| 180 | |
| 181 | |
| 182 | ########################################################################## |