TensorFlow函数：tf.linalg.expm

tf.linalg.expm函数

tf.linalg.expm(
    input,
    name=None
)

计算一个或多个方阵的矩阵指数：

exp(A)= \ sum_ {n = 0} ^ \ infty A ^ n / n！

可以使用缩放和平方法和Pade近似的组合来计算指数.

输入是一个形状为[..., M, M]的张量,其最内部的2个维度形成方形矩阵.输出是与包含所有输入子矩阵的指数的输入相同形状的张量[..., :, :].

参数：

• input：一个Tensor,必须是下列类型之一：float64,float32,complex64,complex128,并且形状是[..., M, M].
• name：操作的名称(可选).

返回：

返回Tensor,与input具有相同的类型.