Produce chunk shapes given a chunk index
| 69 | |
| 70 | |
| 71 | class ArrayChunkShapeDep(ArrayBlockwiseDep): |
| 72 | """Produce chunk shapes given a chunk index""" |
| 73 | |
| 74 | def __getitem__(self, idx: tuple[int, ...]): |
| 75 | return tuple(chunk[i] for i, chunk in zip(idx, self.chunks)) |
| 76 | |
| 77 | |
| 78 | class ArraySliceDep(ArrayBlockwiseDep): |
no outgoing calls
searching dependent graphs…