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

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