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

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