文件名称：broadcast_pycuda
文件大小：9KB
文件格式：ZIP
更新时间：2024-03-02 04:17:35
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)