XLA如何表达matrix_solve

我浏览了XLA的文档，无法找到表达更复杂的矩阵运算的好方法，如matrix_solve，matrix_triangular_solve，cholesky等。 XLA如何处理这个问题？我知道有一个叫做“CallCustom”的追捕操作，但只是想知道更好的方法。XLA如何表达matrix_solve

来源

2017-02-25 Minjie Wang

一般而言，其意图是在常规TensorFlow中指定实际计算。然后，通过即时编译（https://www.tensorflow.org/performance/xla/jit）或前期编译（.../xla/tfcompile）来打开XLA。

对于矩阵解算器的基础支持，注意除了典型的密集矩阵运算之外，XLA还支持一些控制流原语。请参阅https://www.tensorflow.org/performance/xla/operation_semantics关注while循环结构（#while），以及如何从不同选项（#select）中选择输出。

我还没有计算出这是否会产生一个很好的结果，但在高层次看来，它似乎是最基本的部分。

（对不起，缩写链接;我似乎无法发布超过2）

来源

2017-03-01 07:35:56

XLA如何表达matrix_solve

回答

相关问题