mindspore.ops.eigvals

mindspore.ops.eigvals(A)[source]

Computes the eigenvalues of a square matrix(batch square matrices).

Warning

This is an experimental API that is subject to change or deletion.

Parameters

A (Tensor) – Square matrices of shape \((*, N, N)\), with float32, float64, complex64 or complex128 data type.

Returns

Tensor, with shape \((*, N)\). Returns the eigenvalues of the corresponding matrix, which may not have an order.

Raises
  • TypeError – If dtype of A is not one of: float64, float32, complex64 or complex128.

  • TypeError – If A is not a Tensor.

  • ValueError – If A is not a square(batch squares).

Supported Platforms:

Ascend CPU

Examples

>>> 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]