Tensorflow XLA快速矩阵求解
我浏览了XLA的文档,没有找到一个好方法来表达更复杂的矩阵运算,比如矩阵解、矩阵三角解、cholesky等等。XLA如何处理这个?我知道有一个catch call操作CallCustom,但我只是想知道更好的方法。一般来说,目的是在常规TensorFlow中指定实际计算。然后,您可以通过即时编译或提前编译…X/La/TfCopy.打开XLA。Tensorflow XLA快速矩阵求解,tensorflow,Tensorflow,我浏览了XLA的文档,没有找到一个好方法来表达更复杂的矩阵运算,比如矩阵解、矩阵三角解、cholesky等等。XLA如何处理这个?我知道有一个catch call操作CallCustom,但我只是想知道更好的方法。一般来说,目的是在常规TensorFlow中指定实际计算。然后,您可以通过即时编译或提前编译…X/La/TfCopy.打开XLA。 关于矩阵求解器的底层支持,请注意,除了典型的密集矩阵运算之外,XLA确实支持一些控制流基元。请参见关注while循环构造while,以及如何从不同的选择中
关于矩阵求解器的底层支持,请注意,除了典型的密集矩阵运算之外,XLA确实支持一些控制流基元。请参见关注while循环构造while,以及如何从不同的选择中选择输出
我还没有弄清楚这是否会产生一个伟大的结果,但在高层次上,似乎有基本的部分 抱歉的缩写链接;我似乎不能发布超过2个