Returns a dictionary mapping between each index specified in `indices` and the number of output blocks for that indice.
(indices, numblocks, new_axes)
| 1502 | |
| 1503 | |
| 1504 | def _make_dims(indices, numblocks, new_axes): |
| 1505 | """Returns a dictionary mapping between each index specified in |
| 1506 | `indices` and the number of output blocks for that indice. |
| 1507 | """ |
| 1508 | dims = broadcast_dimensions(indices, numblocks) |
| 1509 | for k, v in new_axes.items(): |
| 1510 | dims[k] = len(v) if isinstance(v, tuple) else 1 |
| 1511 | return dims |
| 1512 | |
| 1513 | |
| 1514 | def fuse_roots(graph: HighLevelGraph, keys: list): |
no test coverage detected