为PyCUDA GPUArray启用广播 产品特点 扩展pycuda.elementwise.Elementwise类以启用广播 扩展pycuda.gpuarray.GPUArray以使用修改后的Elementwise 它是如何工作的 您可以在找到此示例 从src导入ElementwiseKernel并像这样使用 ElementwiseKernel ( "double *out, double *a, double *b, double *c" , "out[i] = sin(a