Skip to content

↔ [Converter] Add support for aten::einsum in TRTorch #277

Closed
@narendasan

Description

@narendasan

aten::einsum

  • Function Schema:

aten::einsum(str equation, Tensor[] tensors) -> (Tensor)

  • Original PyTorch API:

https://pytorch.org/docs/stable/generated/torch.einsum.html?highlight=einsum#torch.einsum

  • Relevant TensorRT Documentation:

Alternatives

Additional context

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