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

Python mxnet.ndarray.utils.zeros用法及代碼示例

用法:

mxnet.ndarray.utils.zeros(shape, ctx=None, dtype=None, stype=None, **kwargs)

參數：

  • shape：(int or tuple of int) - 空數組的形狀
  • ctx：(mxnet.context.Context, optional) - 一個可選的設備上下文(默認是當前的默認上下文)
  • dtype：(str or numpy.dtype, optional) - 可選值類型(默認為float32)
  • stype：(string, optional) - 空數組的存儲類型，如‘row_sparse’, ‘csr’等

返回：

創建的數組

返回類型：

NDArray、CSRNDArray或RowSparseNDArray

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

例子：

>>> mx.nd.zeros((1,2), mx.cpu(), stype='csr')
<CSRNDArray 1x2 @cpu(0)>
>>> mx.nd.zeros((1,2), mx.cpu(), 'float16', stype='row_sparse').asnumpy()
array([[ 0.,  0.]], dtype=float16)

相關用法


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