Warning: file_get_contents(/data/phpspider/zhask/data//catemap/0/performance/5.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
Arrays Julia中的数组/张量计算效率_Arrays_Performance_Optimization_Julia_Tensor - Fatal编程技术网

Arrays Julia中的数组/张量计算效率

Arrays Julia中的数组/张量计算效率,arrays,performance,optimization,julia,tensor,Arrays,Performance,Optimization,Julia,Tensor,我想知道Julia中最有效的数组计算方法是什么。我想从头开始写一些深入的学习模型,所以也许有一些用于张量计算的软件包,也许在gpu上? 我当前的代码非常基本: function linear(x, w, b) return(x*w .+ transpose(b)) end 好消息:这段代码已经很好地实现了最大性能和gpu。您只需使用CuArray(对于Nvidia)来定义GPU阵列类型,然后就可以运行linear(CuArray(1:1000)、CuArray(1:2*1:1000')

我想知道Julia中最有效的数组计算方法是什么。我想从头开始写一些深入的学习模型,所以也许有一些用于张量计算的软件包,也许在gpu上? 我当前的代码非常基本:

function linear(x, w, b)
    return(x*w .+ transpose(b))
end

好消息:这段代码已经很好地实现了最大性能和gpu。您只需使用
CuArray
(对于Nvidia)来定义GPU阵列类型,然后就可以运行
linear(CuArray(1:1000)、CuArray(1:2*1:1000')、CuArray(1:1000))
,所有计算都将在GPU上进行。请注意,您可能需要更复杂的示例,以使GPU速度值得数据传输时间