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

Question

我想知道 Julia 中最有效的数组计算方式是什么。我想从头开始编写一些深度学习模型，所以也许有一些用于张量计算的包，也许在 gpu 上？我当前的代码几乎是基本的：

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

标签： arraysperformanceoptimizationjuliatensor

Answer 1

好消息：这段代码对于最大化性能和 gpu 来说已经很好了。你只需要一个CuArrays（对于 Nvidia）来定义一个 GPU 数组类型，然后你可以运行类似的代码linear(CuArray(1:1000),CuArray(1:2*1:1000'),CuArray(1:1000))，所有的计算都将在 GPU 上发生。请注意，您可能需要更复杂的示例才能使 GPU 速度值得数据传输时间。