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