mindspore.ops.eigvals

查看源文件
mindspore.ops.eigvals(A)[源代码]

计算输入方阵(batch方阵)的特征值。

警告

这是一个实验性API,后续可能修改或删除。

参数:
  • A (Tensor) - 方阵。shape为 \((*, N, N)\),数据类型支持 float32、float64、complex64、complex128。

返回:
  • eigen_values (Tensor) - shape为 \((*, N)\),其中的每个向量代表对应矩阵的特征值,特征值之间没有顺序关系。

异常:
  • TypeError - 如果 A 的数据类型不是:float32、float64、complex64或者complex128。

  • TypeError - 如果 A 不是一个Tensor。

  • ValueError - 如果 A 不是一个方阵(batch方阵)。

支持平台:

Ascend CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> import numpy as np
>>> input_x = Tensor(np.array([[1.0, 0.0], [0.0, 2.0]]), mindspore.float32)
>>> u = ops.eigvals(input_x)
>>> print(u)
[1.+0.j 2.+0.j]