megengine.functional.arange

arange(start=0, stop=None, step=1, dtype='float32', device=None)[源代码]

返回一个数值从 startstop ,相邻间隔为 step 的张量。

参数
  • start – 序列的第一个值,应为标量。

  • stop – 序列的最后一个值,应为标量。

  • step – 每对相邻值之间的间隔。默认:1

  • dtype – 结果的数据类型。

返回

生成的张量。

例如:

import numpy as np
import megengine.functional as F

a = F.arange(5)
print(a.numpy())

输出:

输出:

[0. 1. 2. 3. 4.]