是否有任何机制可以在 Matlab / Octave 中自动挤压

Question

对于 nD 数组，能够自动挤压以删除单维数会很好。有没有办法做到这一点，我不知道？这对于聚合函数（例如求和、均值等）特别有用，在这些函数中，您总是希望得到具有较少维度的结果。

这是一个简单的例子：

>> A = ones(3,3,3);
>> B = mean(A);
>> size(B)
ans =  
   1   3   3

>> squeeze(B)
ans =
   1   1   1
   1   1   1
   1   1   1

如果 Matlab/Octave 能够自动为我进行挤压，那就太好了。或者，如果有办法打开该选项（类似于 hold on 的情节）。

Answer 1

据我所知，Matlab 没有。而且我认为这不是一个好主意。考虑您的示例的修改版本：

>> A = ones(3,1,1,3);
>> B = mean(A);
>> size(B)
ans =
     1     1     1     3

“自动挤压”在这里应该做什么？将B 缩小为[1 1 3] 或[1 3]？

• 您可以争辩说它应该删除mean 已变成单例的相同 维度。但它必须在mean 函数内 完成，可能带有一个可选的输入参数。获得函数输出后，没有任何信息如何获得它。

• 或者您可以争辩说它应该删除 all 单例维度，就像squeeze（或多或少）那样。但随后它会删除函数 input 中已经是单例的维度，这可能是不需要的。

如果你问我，在squeeze 中输入 second 指定要删除的（单个）尺寸将是一个很好的补充（与您可以使用 mean(A, 1) 强制即使A 恰好是行向量，也要沿第一个维度应用的操作）。

Answer 2

我同意 Luis 和 Cris 的观点，但我会补充以下内容。

Matlab 和 Octave do 都会自动挤压额外的维度，在一个非常特殊的场景中：在末尾​​em> 已被简化为单例的任何维度都会被自动挤压出来。

例如

A = ones([1,2,3,4]);
B = mean(A, 4);
size(B)
% ans = 1   2   3

注意，答案是[1,2,3]，而不是 [1,2,3,1]。这与诸如 python 之类的语言形成对比，例如，(1,1) 的大小与 (1,) 的大小非常不同。

因此，关于您的问题，利用这一点对您有利的一种方法可能是确保始终在最后找到要减少的维度，从而自动简化。

当您意识到这一点时，这将变得更加有用：

size(A(:))       % ans =   24    1     (i.e. 24)
size(A(:,:))     % ans =    1   24
size(A(:,:,:))   % ans =    1    2   12
size(A(:,:,:,:)) % ans =    1    2    3    4

意思是，如果您对维度进行分层排序，您可以确保任何需要在更高维度上进行的操作，a) 可以轻松矢量化，并且 b) 给出自然的结果，而无需浪费时间挤压或排列结果维度。