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

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) 的數組共享數據。

返回類型：

ND陣列

在不改變任何數據的情況下向當前數組添加一個額外的維度。

例子：

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