减少矩阵乘法中的分配次数?

问题描述 投票:0回答:1

在 Julia 中矩阵相乘时有什么方法可以减少分配次数吗?如果可能的话,我想减少这个数字。我已附上显示此内容的屏幕截图。

matrix julia linear-algebra numerical-methods
1个回答
3
投票

如果预先分配输出矩阵,则可以减少到不分配:

julia> x = rand(600, 600);

julia> y = rand(600, 600);

julia> z = zeros(600, 600);

julia> @allocated mul!(z, x, y)
0
© www.soinside.com 2019 - 2024. All rights reserved.