Method choice

dask/array/random.py:128–162

(
        self,
        a,
        size=None,
        replace=True,
        p=None,
        axis=0,
        shuffle=True,
        chunks="auto",
    )

    @derived_from(np.random.Generator, skipblocks=1)
    def choice(
        self,
        a,
        size=None,
        replace=True,
        p=None,
        axis=0,
        shuffle=True,
        chunks="auto",
    ):
        (
            a,
            size,
            replace,
            p,
            axis,
            chunks,
            meta,
            dependencies,
        ) = _choice_validate_params(self, a, size, replace, p, axis, chunks)

        sizes = list(product(*chunks))
        bitgens = _spawn_bitgens(self._bit_generator, len(sizes))

        name = "da.random.choice-%s" % tokenize(
            bitgens, size, chunks, a, replace, p, axis, shuffle
        )
        keys = product([name], *(range(len(bd)) for bd in chunks))
        dsk = {
            k: Task(k, _choice_rng, bitgen, a, size, replace, p, axis, shuffle)
            for k, bitgen, size in zip(keys, bitgens, sizes)
        }

        graph = HighLevelGraph.from_collections(name, dsk, dependencies=dependencies)
        return Array(graph, name, chunks, meta=meta)

    @derived_from(np.random.Generator, skipblocks=1)
    def exponential(self, scale=1.0, size=None, chunks="auto", **kwargs):