Julia dropdims用法及代碼示例

dropdims(A; dims)

返回一個與 A 具有相同數據的數組，但刪除了 dims 指定的維度。 size(A,d) 對於 dims 中的每個 d 必須等於 1，並且禁止在 1:ndims(A) 之外重複維度或數字。

結果與 A 共享相同的基礎數據，因此當且僅當 A 是可變的並且設置一個元素的元素會改變另一個的值時，結果才是可變的。

另請參閱： reshape 、 vec 。

例子

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

julia> b = dropdims(a; dims=3)
2×2×1 Array{Int64, 3}:
[:, :, 1] =
 1  3
 2  4

julia> b[1,1,1] = 5; a
2×2×1×1 Array{Int64, 4}:
[:, :, 1, 1] =
 5  3
 2  4