三角形矩阵矩阵在TensorFlow中乘以trmm

时间:2019-05-16 14:00:30

标签: tensorflow linear-algebra blas cublas

对于其中一个矩阵为下三角的情况,我需要在TF中获得最快的matmul操作。 cuBLAS和BLAS具有trmm功能,但TensorFlow似乎无法从中受益。

我检查了LowerOperators的LinearOperators实现。但是,尚不清楚它是否利用BLAS实现。

谁能确认大多数优化版本是由LinearOperators实现的?

谢谢!

0 个答案:

没有答案