我浏览了XLA的文档，无法找到表达更复杂的矩阵运算的好方法，如matrix_solve，matrix_triangular_solve，cholesky等。 XLA如何处理这个？我知道有一个接听电话操作“CallCustom”，但只是想知道更好的方法。
答案 0 :(得分：1)
通常，目的是在常规TensorFlow中指定实际计算。然后通过即时编译（https://www.tensorflow.org/performance/xla/jit）或Ahead-Of-Time编译（... / xla / tfcompile）打开XLA。
就矩阵求解器的底层支持而言，请注意除了典型的密集矩阵运算外，XLA确实支持一些控制流原语。请参阅https://www.tensorflow.org/performance/xla/operation_semantics注意while循环结构（#while），以及如何从不同的选择中选择输出（#select）。
我还没有弄清楚这是否会产生很好的结果，但从高层次来看，基本的部分似乎就在那里。
（对不起，缩写链接;我似乎不能发布超过2个）