bigmultiplier的Python项目详细描述
为矩阵乘法创建的存储库。
目的是防止GPU内存不足。
做矩阵乘法运算是CPU的一个大运算。然而,一般来说,gpu ram是不够的。有时无法将矩阵中的两个放在gpu ram上。
在这个实现中,您将发现一个theano符号乘法。这个函数将把你的大矩阵划分成更小的块,以便在gpu上进行计算。
只需调用方法
result_sparse_matrix = calculate(sparse_matrix_1, sparse_matrix_2)
实现使用的是scipy-lil和csr稀疏矩阵。
安装
pip install bigmultiplier
欢迎加入QQ群-->: 979659372
推荐PyPI第三方库