megengine.functional.linspace

linspace(start, stop, num, dtype='float32', device=None)[源代码]

返回指定间隔的等距数列。

参数

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

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

  • num (Union[int, Tensor]) – 将要产生的值的个数。

  • dtype – 结果的数据类型。

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

返回类型

Tensor

返回

生成的张量。

例如：

import numpy as np
import megengine.functional as F

a = F.linspace(3, 10, 5)
print(a.numpy())

输出：

[ 3.    4.75  6.5   8.25 10.  ]
megengine.functional.arange megengine.functional.eye