阅读(6.6k) 书签 (0)

TensorFlow函数:tf.linalg.expm

2018-08-10 10:41 更新

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具有相同的类型.