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

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


用法:

flatten(inplace=False)

参数：

inplace：(bool, default False) - 如果为 True，此方法返回一个看法与该数组共享数据的该数组。否则，将返回一个副本。

返回：

具有扁平形状 (d1, d2*…*dk) 的数组，它与形状为 (d1, d2, …, dk) 的该数组共享数据。

返回类型：

ND阵列

展平此数组而不更改任何数据。

例子：

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

相关用法


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