一个帮助矩阵乘法的包

bigmultiplier的Python项目详细描述


https://travis-ci.org/somedanalytics/big-multiplier.svg?branch=master

为矩阵乘法创建的存储库。

目的是防止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 Python中文网_新手群

推荐PyPI第三方库


热门话题
开始编码器(Java)验证字符串输入,并在输入正确的输入后结束循环   java如何修复请求属性的跨站点脚本?   java打印密码生成器   带有数组或字符串的java JSON解析实例类型   java通过类使用jsoup从页面获取值   java如何更改微调器后面的背景?   java如何在libgdx(风险游戏)中定义可高亮显示、可单击的自定义对象   java反射式地创建一个匿名类   java如何在Spring应用程序(类内)中处理国际化   java获取“ShortenedMonth/dd”格式的日期   java菱形运算符ArrayList   java如何按列和行添加到for循环中的矩形数组   在Java中查找数组中彼此距离最小的两个数字   JAVA中JTextField内的swing设置IP地址   java如何更新代码布局   带Mac的java开放JAR