Julia mapslices用法及代码示例

mapslices(f, A; dims)

使用函数 f 转换数组 A 的给定维度。 f 在 A 的每个切片上调用 A[...,:,...,:,...] 。 dims 是一个整数向量，用于指定冒号在此表达式中的位置。结果沿其余维度连接。例如，如果 dims 是 [1,2] 并且 A 是 4 维的，则在 A[:,:,i,j] 上为所有 i 和 j 调用 f。

另见 eachcol 、 eachslice 。

例子

julia> a = reshape(Vector(1:16),(2,2,2,2))
2×2×2×2 Array{Int64, 4}:
[:, :, 1, 1] =
 1  3
 2  4

[:, :, 2, 1] =
 5  7
 6  8

[:, :, 1, 2] =
  9  11
 10  12

[:, :, 2, 2] =
 13  15
 14  16

julia> mapslices(sum, a, dims = [1,2])
1×1×2×2 Array{Int64, 4}:
[:, :, 1, 1] =
 10

[:, :, 2, 1] =
 26

[:, :, 1, 2] =
 42

[:, :, 2, 2] =
 58