当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python mxnet.ndarray.NDArray.expand_dims用法及代码示例

用法: expand_dims(axis, inplace=False)

参数：

axis：(int) - 要插入新轴的位置。假设输入 NDArray 的维度为 ndim，则插入轴的范围为 [-ndim, ndim]。
inplace：(bool, default False) - 如果为 True，此方法返回一个看法与该数组共享数据的该数组。否则，将返回一个副本。

返回：

在给定输入轴 i 的情况下，形状为 (d1, d2, …, 1, di, …, dk) 的数组与形状为 (d1, d2, …, dk) 的数组共享数据。

返回类型：

在不改变任何数据的情况下向当前数组添加一个额外的维度。

例子：

>>> x = mx.nd.arange(6).reshape(2,3)
>>> y = x.expand_dims(1, inplace=True)
>>> z = x.expand_dims(1)
>>> y.shape
(2, 1, 3)
>>> y[0].asnumpy()
array([[0., 1., 2.]], dtype=float32)
>>> y[:] = -1
>>> x.asnumpy()
array([[-1., -1., -1.],
       [-1., -1., -1.]], dtype=float32)
>>> z[0].asnumpy()
array([[0., 1., 2.]], dtype=float32)

相关用法

注：本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.NDArray.expand_dims。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。