julia - 数组维度上的广播

Question

假设我有一个二维数组

a = [1 2 3; 1 2 3]

2×3 Array{Int64,2}:
 1  2  3
 1  2  3

我想要 sum沿着一个维度，例如沿维度 1 屈服

[2, 4, 6]

或沿维度 2 屈服

[6, 6]

这在 Julia 中是如何正确完成的?

Answer 1

julia> sum(a; dims=1)
1×3 Array{Int64,2}:
 2  4  6

julia> sum(a; dims=2)
2×1 Array{Int64,2}:
 6
 6

您可以使用 vec 删除维度.