如何通过第3个dim合并两个4个dims数组

Question

也就是说，将大小为[2,3,4,5]的A和大小为[2,3,6,5]的B合并为大小为[2,3,10,5]的C。

我知道该怎么做。但是如何为我和机器高效地完成它。

我尝试了以下代码，但是没有用。

permutedims([permutedims(A,[1,2,4,3]) permutedims(B,[1,2,4,3])],[1,2,4,3])

Answer 1

您可以为此使用cat（并置）功能。它带有关键字参数dims，可用于指定要连接的尺寸。

julia> A = rand(2,3,4,5);

julia> B = rand(2,3,6,5);

julia> C = cat(A, B, dims=3);

julia> size(C)
(2, 3, 10, 5)

通过在Julia REPL中键入cat，可以找到有关?cat函数的更多信息。