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 函數 range 和 numpy.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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。

用法:

參數：

返回：

返回類型：

例子：