(a, indices, axis)
| 2014 | |
| 2015 | |
| 2016 | def _take_dask_array_from_numpy(a, indices, axis): |
| 2017 | assert isinstance(a, np.ndarray) |
| 2018 | assert isinstance(indices, Array) |
| 2019 | |
| 2020 | return indices.map_blocks( |
| 2021 | lambda block: np.take(a, block, axis), chunks=indices.chunks, dtype=a.dtype |
| 2022 | ) |
| 2023 | |
| 2024 | |
| 2025 | @derived_from(np) |
no test coverage detected