mindspore.ops.matrix_exp

mindspore.ops.matrix_exp(x)

Computes the matrix exponential of a square matrix. Supports batched inputs.

$$matrix\mathrm{\_}exp(x)=\sum _{k=0}^{\mathrm{\infty }}\frac{1}{k!}{x}^k\in {\mathbb{K}}^{n\times n}$$

Parameters

x (Tensor) – The shape of tensor is $(\ast ,n,n)$ where * is zero or more batch dimensions. Must be one of the following types: float16, float32, float64, complex64, complex128.

Returns

Tensor, has the same shape and dtype as the x.

Raises

• TypeError – If x is not a Tensor.

• TypeError – If the dtype of x is not one of the following dtype: float16, float32, float64, complex64, complex128.

• ValueError – If the rank of x is less than 2.

• ValueError – If the last two dimensions of x are not equal.

Supported Platforms:

CPU

Examples

>>> x = Tensor(np.array([[1, 2], [0, 1]]), mindspore.float32)
>>> output = ops.matrix_exp(x)
>>> print(output)
[[2.7182817 5.436563 ]
[0.        2.7182817]]