Skip to content

Link to torch.einsum in torch.tensordot #50802

@shaibagon

Description

@shaibagon

📚 Documentation

Currently the documentation of torch.tensordot is quite comprehensible.
However, I came across several instances recently (e.g., a, b, c) where users tried to "push" tensordot beyond its capabilities while being completely unaware of the functionality provided by torch.einsum.

Is it possible to add "see also" part at the bottom of tensordot documentation linking to einsum?

For example, in numpy.tensordot documentation there is a "see also" part that links to einsum:

image

Can we have similar section in pytorch as well, please?

cc @jlin27 @mruberry

Metadata

Metadata

Assignees

No one assigned

    Labels

    module: docsRelated to our documentation, both in docs/ and docblockstriagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions