(self)
| 294 | |
| 295 | @cached_property |
| 296 | def chunks(self): |
| 297 | chunks = [ |
| 298 | ( |
| 299 | tuple(1 for p in partition_all(self.split_every[i], c)) |
| 300 | if i in self.split_every |
| 301 | else c |
| 302 | ) |
| 303 | for (i, c) in enumerate(self.array.chunks) |
| 304 | ] |
| 305 | |
| 306 | if not self.keepdims: |
| 307 | out_axis = [i for i in range(self.array.ndim) if i not in self.split_every] |
| 308 | getter = lambda k: get(out_axis, k) |
| 309 | chunks = list(getter(chunks)) |
| 310 | return tuple(chunks) |
| 311 | |
| 312 | def _layer(self): |
| 313 | x = self.array |