Skip to content

A protocol for tensordot #11128

@mrocklin

Description

@mrocklin

Following on #11074 I would like to see operations like np.tensordot develop a protocol that could be implemented by duck-array libraries like dask, cupy, or sparse

In [1]: import numpy as np

In [2]: import dask.array as da

In [3]: x = da.ones((5, 5), chunks=(2, 2))

In [4]: np.tensordot(x, x)  # I would like for this to be a dask.array
Out[4]: array(25.)

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions