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

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


用法:

mxnet.ndarray.full(shape, val, ctx=None, dtype=<class 'numpy.float32'>, out=None)

參數：

  • shape：(int or tuple of int) - 新數組的形狀。
  • val：(scalar) - 填充值。
  • ctx：(mxnet.context.Context, optional) - 設備上下文(默認為當前默認上下文)。
  • dtype：(str或者numpy.dtype, 可選) - 返回的數據類型NDArray.默認數據類型是float32.
  • out：(NDArray, optional) - 輸出 NDArray(默認為None)。

返回：

NDArray 填充有 val ，具有給定的形狀、ctx 和 dtype。

返回類型：

ND陣列

返回給定形狀和類型的新數組，填充給定值 val 。

例子：

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

相關用法


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