Skip to content

Support torch.linalg.cross #62810

@AnirudhDagar

Description

@AnirudhDagar

🚀 Feature

Support torch.linalg.cross.

cc @jianyuh @nikitaved @pearu @mruberry @heitorschueroff @walterddr @IvanYashchuk @xwang233 @lezcano @rgommers @pmeier @asmeurer @leofang @AnirudhDagar @asi1024 @emcastillo @kmaehashi

Motivation & Pitch

Python Array API proposes the use of linalg.cross among other linear algebra functions. Currently, PyTorch supports the same functionality with torch.cross. Array-API Ops tracker (#58742) lists a few aliases and it would be nice to add an alias torch.linalg.cross to torch.cross making it compliant with Array API.

Additional context

Metadata

Metadata

Assignees

Labels

module: linear algebraIssues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmulmodule: python array apiIssues related to the Python Array APItriagedThis 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