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