|
|
1
2
我不知道是否存在对矩阵执行元素操作的库,但您可以很容易地设置CUDA内核来完成这项工作。例如,您可以给出
编辑 :看起来您可以使用 Thrust 和宏集 Newton ,如图所示 SO question . |