动态创建 d 维张量

我想使用创建一个 d 维张量d作为输入并且没有if声明如下：

if d == 2
   B = zeros(r,r);
   for i = 1:r
       B(i,i) = 1;
   end                
elseif d == 3
   B = zeros(r,r,r);
   for i = 1:r
       B(i,i,i) = 1;
   end
end

有更有效的方法吗？

您可以使用累加数组 https://www.mathworks.com/help/matlab/ref/accumarray.html:

f = @(d,r)accumarray(repmat((1:r).',1 , d), 1);

> f(2,5)
=
1   0   0   0   0
0   1   0   0   0
0   0   1   0   0
0   0   0   1   0
0   0   0   0   1

这是accumarray 的基本签名：

accumarray( subs , val )

Using accumarray我们可以创建一个 n 维数组，其中subs表示将在数组中填充的点的位置val代表了他们的价值。

If subs作为矩阵提供，其列数决定了结果数组的维数，每行代表每个点的位置。

例如对于d = 2 and r = 5我们想创建一个(5 x 5)在以下位置有 1 的数组：[1 1],[2 2],[3 3],[4 4],[5 5].

Using repmat我们可以创造subs:

subs = repmat ((1:5).' , 1, 2)

=
1 1
2 2
3 3
4 4
5 5

val被设定为1所以所有指定的职位都将由1.

. https://codegolf.stackexchange.com/questions/127904/n-dimensional-identity-matrix/127916#127916