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

Question

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

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

Answer 1

好消息:这段代码已经有利于最大化性能和 gpu。您只需要一个 CuArrays (对于 Nvidia)定义一个 GPU 数组类型，然后你可以运行像 linear(CuArray(1:1000),CuArray(1:2*1:1000'),CuArray(1:1000)) 这样的代码。并且所有计算都将在 GPU 上进行。请注意，您可能需要更复杂的示例来证明 GPU 速度值得花费数据传输时间。