Array/Tensor Julia 的计算效率
Array/Tensor calculations efficiency in Julia
我想知道 Julia 中最有效的数组计算方式是什么。我想从头开始编写一些深度学习模型，所以也许有一些用于张量计算的包，也许在 gpu 上？
我当前的代码非常基本：
function linear(x, w, b)
return(x*w .+ transpose(b))
end
好消息：此代码已经适用于最大性能和 gpu。您只需要 CuArrays（对于 Nvidia）来定义 GPU 数组类型，然后您可以 运行 编写 linear(CuArray(1:1000),CuArray(1:2*1:1000'),CuArray(1:1000)) 之类的代码，所有计算都将在 GPU 上进行。请注意，您可能需要更复杂的示例才能使 GPU 速度值得数据传输时间。
我想知道 Julia 中最有效的数组计算方式是什么。我想从头开始编写一些深度学习模型，所以也许有一些用于张量计算的包，也许在 gpu 上？ 我当前的代码非常基本：
function linear(x, w, b)
return(x*w .+ transpose(b))
end
好消息：此代码已经适用于最大性能和 gpu。您只需要 CuArrays（对于 Nvidia）来定义 GPU 数组类型，然后您可以 运行 编写 linear(CuArray(1:1000),CuArray(1:2*1:1000'),CuArray(1:1000)) 之类的代码，所有计算都将在 GPU 上进行。请注意，您可能需要更复杂的示例才能使 GPU 速度值得数据传输时间。