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

Python mxnet.ndarray.ones用法及代碼示例

用法:

mxnet.ndarray.ones(shape, ctx=None, dtype=None, **kwargs)

參數：

  • shape：(int or tuple of int or list of int) - 空數組的形狀。
  • ctx：(mxnet.context.Context, optional) - 一個可選的設備上下文。默認為當前默認上下文 (mxnet.context.current_context())。
  • dtype：(str or numpy.dtype, optional) - 可選值類型(默認為float32)。
  • out：(NDArray, optional) - 輸出 NDArray(默認為None)。

返回：

指定形狀的新數組，全部填充。

返回類型：

ND陣列

返回一個新的數組，其中填充了所有的數組，具有給定的形狀和類型。

例子：

>>> mx.nd.ones(1).asnumpy()
array([ 1.], dtype=float32)
>>> mx.nd.ones((1,2), mx.gpu(0))
<NDArray 1x2 @gpu(0)>
>>> mx.nd.ones((1,2), dtype='float16').asnumpy()
array([[ 1.,  1.]], dtype=float16)

相關用法


注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.ndarray.ones。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。