當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。