(chunks, axis1, axis2)
| 705 | return axis |
| 706 | |
| 707 | def pop_axes(chunks, axis1, axis2): |
| 708 | chunks = list(chunks) |
| 709 | chunks.pop(axis2) |
| 710 | chunks.pop(axis1) |
| 711 | return tuple(chunks) |
| 712 | |
| 713 | axis1 = _axis_fmt(axis1, "axis1", a.ndim) |
| 714 | axis2 = _axis_fmt(axis2, "axis2", a.ndim) |