当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。