Эффективность вычислений массива / тензора в Юлии

Question

Мне стало интересно, какой самый эффективный способ вычисления массивов в Юлии. Я хочу написать несколько моделей глубокого обучения с нуля, так что, возможно, есть какой-то пакет для тензорных вычислений, может быть, на GPU? Мой текущий код довольно похож на basi c, так как он получает:

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

Answer 1

Хорошие новости: этот код уже хорош для максимальной производительности и графического процессора. Вам просто нужно CuArrays (для Nvidia), чтобы определить тип массива GPU, а затем вы можете запустить код, подобный linear(CuArray(1:1000),CuArray(1:2*1:1000'),CuArray(1:1000)), и все вычисления будут выполняться на GPU. Обратите внимание, что вам, вероятно, понадобятся более сложные примеры, чтобы скорость графического процессора стоила времени передачи данных.