文件名称:cudamat:cudamat的私有副本(Windows支持,与Theano和PyCUDA的互操作性)
文件大小:89KB
文件格式:ZIP
更新时间:2024-06-10 07:17:15
Python
库达马特 cudamat项目的目的是使在使用Python的支持CUDA的GPU上执行基本矩阵计算变得容易。 cudamat提供了在GPU上执行计算的Python矩阵类。 目前,我们的GPU矩阵类支持的一些操作包括: 轻松与numpy.ndarray实例之间进行numpy.ndarray 。 切片支持有限。 矩阵乘法和转置。 逐元素加法,减法,乘法和除法。 exp,log,pow,sqrt的元素化应用。 行或列的总和,最大值和最小值。 将CUDA错误转换为Python异常。 cudamat的当前功能集偏向于实现一些常见的机器学习算法所需的功能。 在cudamat随附的示例中,我们包括前馈神经网络和受限Boltzmann机器的实现。 例子: import numpy as np import cudamat as cm cm . cublas_init () # crea
【文件预览】:
cudamat-master
----setup.py(3KB)
----.gitignore(364B)
----INSTALL.md(2KB)
----CHANGELOG(2KB)
----cudamat()
--------learn_kernels.cuh(344B)
--------cudamat_kernels.cuh(6KB)
--------learn.py(857B)
--------__init__.py(43B)
--------cudamat.cuh(620B)
--------rnd_multipliers_32bit.txt(172KB)
--------Makefile.win(336B)
--------cudamat_kernels.cu(25KB)
--------learn.cu(841B)
--------learn_kernels.cu(374B)
--------cudamat.cu(46KB)
--------cudamat.py(45KB)
----LICENSE(1KB)
----examples()
--------bench_cudamat.py(3KB)
--------util.py(614B)
--------nn_cudamat.py(3KB)
--------rbm_cudamat.py(2KB)
--------rbm_numpy.py(2KB)
----README.md(2KB)
----CONTRIBUTE.md(2KB)
----test()
--------test_learn.py(714B)
--------test_cudamat.py(35KB)