matrix - Julia - 如何有效地将矩阵的对角线归零？

Question

在 Julia 中，将矩阵的对角线变为零的有效方法是什么？

Answer 1

假设 m是你的矩阵大小 N x N可以这样做:

setindex!.(Ref(m), 0.0, 1:N, 1:N)

另外一个选项:

using LinearAlgebra
m[diagind(m)] .= 0.0

以及一些性能测试:

julia> using LinearAlgebra, BenchmarkTools

julia> m=rand(20,20);

julia> @btime setindex!.(Ref($m), 0.0, 1:20, 1:20);
  55.533 ns (1 allocation: 240 bytes)

julia> @btime $m[diagind($m)] .= 0.0;
  75.386 ns (2 allocations: 80 bytes)