-
Notifications
You must be signed in to change notification settings - Fork 24.7k
Description
🚀 Feature
Support torch.linalg.outer
.
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.outer
among other linear algebra functions. Currently, PyTorch supports the same functionality with torch.outer
. torch.ger
(to be deprecated in 1.10) is also an alias to torch.outer
at the moment.
We can either make torch.linalg.outer
the main function with torch.outer
and torch.ger
as aliases to that. Or we can keep torch.outer
as is and just alias torch.linalg.outer
to that.
The only difference is that Array API doesn't define a keyword-only out=
argument which should be fine since PyTorch's API will just be a superset of Array API here.