Skip to content

Automatically propagate symmetry to einsum output #37

@wiwu2390

Description

@wiwu2390

Currently, einsum outputs raw WhestArrays. Since we're already computing the symmetries of the output for the purpose of flop counting, we should automatically propagate it to the output.

Example:

A = we.random.randn(n, n, n)
W = we.random.randn(n, n)
X = we.einsum('ijk,ai,bj,ck->abc', A, W, W, W)    # Currently a raw WhestArray. Should instead be a SymmetricTensor with all axes symmetric

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or requestpriority:p1Ship in the current iteration

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions