Julia：关于未确定维度的数组的问题

Question

我有两个初学者的问题：

(1) 我想重塑一个数组，但维度来自一个可以是变量的向量。例如，

A = ones(120,1)
b = [2,3,4,5]

我会写

C = reshape(A,2,3,4,5)

但如果 b 可以变化，我想要类似的东西

C = reshape(A,b)

这段代码在 Matlab 中工作。 Julia 中有类似的吗？

(2) 我想对一个高维数组进行切片，同时保持维度灵活。在上面的例子中，我修复了最后一个维度：

C[:,:,:,1]
C[:,:,:,2]

等等。问题是找到一种有效的方法：对于任意维度的数组，我总是可以固定最后一个维度并提取值。

任何帮助将不胜感激！

Answer 1

(1)C = reshape(A,b...)

(2) EllipsisNotation.jl 提供了一个.. 运算符，所以C[..,1] 做你想做的事。

Answer 2

如果您不想安装软件包，则 (2) 有 C[ntuple(x->:, ndims(C)-1)..., 1]。