megengine.functional.arange

arange(start=0, stop=None, step=1, dtype='float32', device=None)

返回一个数值从 start 到 stop ，相邻间隔为 step 的张量。

参数

• start (Union[int, float, Tensor]) – 序列的第一个值，应为标量。

• stop (Union[int, float, Tensor, None]) – 序列的最后一个值，应为标量。

• step (Union[int, float, Tensor]) – 每对相邻值之间的间隔。默认：1

• dtype – 结果的数据类型。

• device (Optional[megengine.core._imperative_rt.common.CompNode]) –

返回类型

Tensor

返回

生成的张量。

例如：

import numpy as np
import megengine.functional as F

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

输出：

输出：

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

megengine.functional.full_like megengine.functional.linspace