eigvals

paddle.linalg. eigvals ( x, name=None ) [源代码]

计算一个(或一批)普通方阵的特征值。

该 API 的反向实现尚未完成,若你的代码需要对其进行反向传播,请使用 ref:cn_api_linalg_eig

参数

  • x (Tensor)- 需要计算特征值的方阵。输入的 Tensor 维度为 [*, M, M],其中 * 表示矩阵的批次维度。支持 float32float64complex64complex128 四种数据类型。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor,包含 x 的所有未排序特征值。返回的 Tensor 具有与 x 相同的批次维度。即使输入的 x 是实数 tensor,返回的也会是复数的结果。

代码示例

import paddle

paddle.set_device("cpu")
paddle.seed(1234)

x = paddle.rand(shape=[3, 3], dtype='float64')
# [[0.02773777, 0.93004224, 0.06911496],
#  [0.24831591, 0.45733623, 0.07717843],
#  [0.48016702, 0.14235102, 0.42620817]])

print(paddle.linalg.eigvals(x))
# [(-0.27078833542132674+0j), (0.29962280156230725+0j), (0.8824477020120244+0j)] #complex128