KEMBAR78
[numpy] Add torch.newdim/torch.newaxis · Issue #65307 · pytorch/pytorch · GitHub
Skip to content

[numpy] Add torch.newdim/torch.newaxis #65307

@kshitij12345

Description

@kshitij12345

For compatibility with existing numpy programs, it would be nice to have torch.newdim/newaxis.

Note : PyTorch already has torch.unsqueeze and one can use None to get the same functionality.

Thanks @AnirudhDagar for pointing this out.

cc @mruberry @rgommers @heitorschueroff

Metadata

Metadata

Assignees

No one assigned

    Labels

    featureA request for a proper, new feature.module: numpyRelated to numpy support, and also numpy compatibility of our operatorstriagedThis 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