783c109c591c6a3affc750d29c9d56f86c1ed6cb,dask_image/ndmeasure/_utils.py,,_ravel_shape_indices,#,53

Before Change

    Gets the raveled indices shaped like input.
    

    indices = sum([
        dask.array.arange(
            0,
            numpy.prod(dimensions[i:], dtype=dtype),
            numpy.prod(dimensions[i + 1:], dtype=dtype),
            dtype=dtype,
            chunks=c
        )[i * (None,) + (slice(None),) + (len(dimensions) - i - 1) * (None,)]
        for i, c in enumerate(chunks)
    ])

    return indices

After Change

        for i, c in enumerate(chunks)
    ]

    indices = dask.array.atop(
        _ravel_shape_indices_kernel, tuple(range(len(indices))),
        *sum([(a, (i,)) for i, a in enumerate(indices)], tuple()),
        dtype=dtype
    )

    return indices

Instances