eigvals¶
计算一个(或一批)普通方阵的特征值。
该 API 的反向实现尚未完成,若你的代码需要对其进行反向传播,请使用 ref:cn_api_paddle_linalg_eig。
参数¶
x (Tensor)- 需要计算特征值的方阵。输入的 Tensor 维度为
[*, M, M]
,其中*
表示矩阵的批次维度。支持float32
、float64
、complex64
和complex128
四种数据类型。name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。
返回¶
Tensor
,包含 x 的所有未排序特征值。返回的 Tensor 具有与 x 相同的批次维度。即使输入的 x 是实数 tensor,返回的也会是复数的结果。
代码示例¶
>>> import paddle
>>> paddle.seed(2023)
>>> x = paddle.rand(shape=[3, 3], dtype='float64')
>>> print(x)
Tensor(shape=[3, 3], dtype=float64, place=Place(cpu), stop_gradient=True,
[[0.86583615, 0.52014721, 0.25960938],
[0.90525323, 0.42400090, 0.40641288],
[0.97020893, 0.74437359, 0.51785128]])
>>> print(paddle.linalg.eigvals(x))
Tensor(shape=[3], dtype=complex128, place=Place(cpu), stop_gradient=True,
[ (1.788956694280852+0j) , (0.16364484879581526+0j),
(-0.14491322408727625+0j)])