Skip to content

ENH: Add matmul equivalent of multi_dot #8719

Open
@eric-wieser

Description

@eric-wieser

#4977 added np.linalg.multi_dot, before np.matmul existed. However, this only works for single 2D matrices.

It ought to be relatively straightforward to add a similar function that works on stacks of matrices, using np.matmul.

What should such a function be called? Or can multi_dot be internally changed to actually call matmul? Because for the inputs it currently accepts, there would be no visible difference in behaviour

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