A convenience function for operations that want to collapse to a scalar like _align, but are using keepdims=True
(self, axis)
| 257 | raise ValueError("unsupported axis") |
| 258 | |
| 259 | def _collapse(self, axis): |
| 260 | """A convenience function for operations that want to collapse |
| 261 | to a scalar like _align, but are using keepdims=True |
| 262 | """ |
| 263 | if axis is None: |
| 264 | return self[0, 0] |
| 265 | else: |
| 266 | return self |
| 267 | |
| 268 | # Necessary because base-class tolist expects dimension |
| 269 | # reduction by x[0] |