Class GUfuncLeafExpr

dask/array/_array_expr/_gufunc.py:535–577

class GUfuncLeafExpr(ArrayExpr):
    _parameters = [
        "array",
        "i",
        "name_prefix",
        "loop_output_chunks",
        "core_shapes",
        "ocd",
        "nout",
        "input_meta",
        "loop_output_shape",
    ]

    @functools.cached_property
    def _meta(self):
        return meta_from_array(self.input_meta, len(self._shape))

    @functools.cached_property
    def _name(self):
        last_name = self.array._name.split("-")[-1]
        return f"{self.name_prefix}_{self.i}-{last_name}"

    @functools.cached_property
    def _shape(self):
        core_output_shape = tuple(self.core_shapes[d] for d in self.ocd)
        return self.loop_output_shape + core_output_shape

    @functools.cached_property
    def chunks(self):
        output_chunks = self.loop_output_chunks + tuple(
            self.core_shapes[d] for d in self.ocd
        )
        return normalize_chunks(output_chunks, self._shape, dtype=self._meta.dtype)

    def _layer(self):
        core_chunkinds = len(self.ocd) * (0,)
        leaf_dsk = {
            (self._name,)
            + key[1:]
            + core_chunkinds: ((getitem, key, self.i) if self.nout else key)
            for key in list(flatten(self.array.__dask_keys__()))
        }
        return leaf_dsk


class gufunc: