Skip to content

Einops now can use array API to manipulate tensors #648

@arogozhnikov

Description

@arogozhnikov

Hi folks, a small announcement:

Einops just gained support for API-compliant tensors:

from einops.array_api import rearrange, reduce, repeat, pack, unpack
from numpy.array_api import zeros

x = zeros([2, 3, 4, 5])
rearrange(x, 'b h w c -> b c h w').shape

This is already available for testing:

pip install einops==0.7.0rc1

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions