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]