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