有没有一种更快的方法来从一个方阵中移除所有非对角线元素？

我在一个需要从一个大的方阵中移除非对角元素的应用程序中遇到了一个小的性能瓶颈。所以，矩阵 x \n

17    24     1     8    15
23     5     7    14    16
 4     6    13    20    22
10    12    19    21     3
11    18    25     2     9

\n变成了\n

17     0     0     0     0
 0     5     0     0     0
 0     0    13     0     0
 0     0     0    21     0
 0     0     0     0     9

\n问题：到目前为止，下面的bsxfun和diag解决方案是最快的解决方案，我怀疑在保持代码在Matlab中的情况下，我能找到更快的方法吗？\n

解决方案

\n这是我目前想到的解决方案。\n对矩阵进行逐元素的乘法，使用单位矩阵。这是最简单的解决方案：\n

y = x .* eye(n);

\n使用bsxfun和diag：\n

y = bsxfun(@times, diag(x), eye(n));

\n下三角/上三角矩阵：\n

y = x - tril(x, -1) - triu(x, 1);

\n使用循环的各种解决方案：\n

y = x;
for ix=1:n
    for jx=1:n
        if ix ~= jx
            y(ix, jx) = 0;
        end
    end
end

\n和\n

y = x;
for ix=1:n
    for jx=1:ix-1
        y(ix, jx) = 0;
    end
    for jx=ix+1:n
        y(ix, jx) = 0;
    end
end

\n

计时

\n实际上，bsxfun的解决方案是最快的。这是我的计时代码：\n

function timing()
clear all
n = 5000;
x = rand(n, n);
f1 = @() tf1(x, n);
f2 = @() tf2(x, n);
f3 = @() tf3(x);
f4 = @() tf4(x, n);
f5 = @() tf5(x, n);
t1 = timeit(f1);
t2 = timeit(f2);
t3 = timeit(f3);
t4 = timeit(f4);
t5 = timeit(f5);
fprintf('t1: %f s\n', t1)
fprintf('t2: %f s\n', t2)
fprintf('t3: %f s\n', t3)
fprintf('t4: %f s\n', t4)
fprintf('t5: %f s\n', t5)
end
function y = tf1(x, n)
y = x .* eye(n);
end
function y = tf2(x, n)
y = bsxfun(@times, diag(x), eye(n));
end
function y = tf3(x)
y = x - tril(x, -1) - triu(x, 1);
end
function y = tf4(x, n)
y = x;
for ix=1:n
    for jx=1:n
        if ix ~= jx
            y(ix, jx) = 0;
        end
    end
end
end
function y = tf5(x, n)
y = x;
for ix=1:n
    for jx=1:ix-1
        y(ix, jx) = 0;
    end
    for jx=ix+1:n
        y(ix, jx) = 0;
    end
end
end

\n它返回：\n

t1: 0.111117 s
t2: 0.078692 s
t3: 0.219582 s
t4: 1.183389 s
t5: 1.198795 s