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


Python mxnet.ndarray.arange用法及代码示例


用法:

mxnet.ndarray.arange(start, stop=None, step=1.0, repeat=1, infer_range=None, ctx=None, dtype=<class 'numpy.float32'>)

参数

  • start(number, optional) - 间隔开始。默认起始值为 0。
  • stop(number) - 间隔结束。
  • step(number, optional) - 值之间的间距。默认步长为 1。
  • repeat(int, optional) - 重复每个元素的次数。默认重复次数为 1。
  • infer_range(boolean, optional) - 从开始、步进、重复和输出张量大小推断停止位置。已弃用。仅支持 False。
  • ctx(mxnet.context.Context, optional) - 设备上下文。默认上下文是当前的默认上下文。
  • dtype(str or numpy.dtype, optional) - 的数据类型NDArray.默认数据类型是np.float32.

返回

NDArray 在指定范围内均匀分布的值。

返回类型

ND阵列

返回给定间隔内均匀分布的值。

值在半开区间 [ start , stop ) 内生成。换句话说,该区间包括 start 但不包括 stop 。该函数类似于内置的 Python 函数 rangenumpy.arange ,但返回一个 NDArray

例子

>>> mx.nd.arange(3).asnumpy()
array([ 0.,  1.,  2.], dtype=float32)
>>> mx.nd.arange(2, 6).asnumpy()
array([ 2.,  3.,  4.,  5.], dtype=float32)
>>> mx.nd.arange(2, 6, step=2).asnumpy()
array([ 2.,  4.], dtype=float32)
>>> mx.nd.arange(2, 6, step=1.5, repeat=2).asnumpy()
array([ 2. ,  2. ,  3.5,  3.5,  5. ,  5. ], dtype=float32)
>>> mx.nd.arange(2, 6, step=2, repeat=3, dtype='int32').asnumpy()
array([2, 2, 2, 4, 4, 4], dtype=int32)

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.ndarray.arange。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。