Tensorflow矩阵平方根代码源

Question

在哪里可以获取tf.linalg.sqrtm的源代码并可以对其进行修改？

Answer 1

tf.linalg.sqrtm只是matrix_square_root的别名。其源代码托管在github上，可以找到here。

TensorFlow代码是Apache许可下的开源代码，因此，可以肯定的是，如果您知道自己在做什么，则可以对其进行扩展并将其修改为自己的要求。也许，如果您的用例是针对您自己的需求的，那么您也可以只构建自己的替代sqrtm函数。