torch.eig should return complex tensor #43081
Labels
enhancement
Not as big of a feature, but technically not a bug. Should be easy to fix
module: bc-breaking
Related to a BC-breaking change
module: complex
Related to complex number support in PyTorch
module: linear algebra
Issues related to specialized linear algebra operations in PyTorch; includes matrix multiply matmul
triaged
This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Currently the return value of
torch.eig
is:eigenvalues (Tensor):
eigenvectors (Tensor):
This behavior is because of historical reason when complex tensors was not a thing. Since complex tensor is now supported,
the return value of
eigenvalues
should then be a complex tensor of shape(n,)
, andeigenvectors
should also be a complex tensor of shape(n,n)
whereeignvectors[:,j]
is the jth eigen vector.cc @ezyang @anjali411 @dylanbespalko @mruberry @gchanan @vincentqb @vishwakftw @jianyuh @nikitaved @pearu
The text was updated successfully, but these errors were encountered: