Skip to content

ENH: broadcast lstsq #8720

Open
Open
@nschloe

Description

@nschloe

Many linalg functions are already broadcastable, lstsq isn't.

A workaround is via svd which is already broadcasted:

u, s, v = numpy.linalg.svd(A, full_matrices=False)
uTb = numpy.einsum('ijk,ij->ik', u, b)
xx = numpy.einsum('ijk, ij->ik', v, uTb / s)

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