megengine.functional.arange

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

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

参数
返回类型

Tensor

返回

生成的张量。

例如:

import numpy as np
import megengine.functional as F

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

输出:

输出:

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