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]