【文件属性】：
文件名称：broadcast_pycuda
文件大小：9KB
文件格式：ZIP
更新时间：2021-02-10 10:30:55
Python
为PyCUDA GPUArray启用广播
产品特点
扩展pycuda.elementwise.Elementwise类以启用广播
扩展pycuda.gpuarray.GPUArray以使用修改后的Elementwise
它是如何工作的
您可以在找到此示例
从src导入ElementwiseKernel并像这样使用
ElementwiseKernel (
"double *out, double *a, double *b, double *c" ,
"out[i] = sin(a[i] * b[i]) + sqrt(c[i]) + cos(a[i]) / log(cos(b[i]) + 2)" ,
broadcasting = True # or False
)
broadcasting=False时的实际内核功能（默认情况下， broadcasting为Fal
【文件预览】：
broadcasting_pycuda-master
----README.md(2KB)
----test()
--------test_broadcasting.py(5KB)
----LICENSE(89B)
----src()
--------broadcast.pyi(348B)
--------broadcast.py(4KB)
--------modified_elementwise.py(7KB)
--------__init__.py(156B)
--------modified_gpuarray.py(4KB)
----CHANGELOG.txt(188B)