mindspore.ops.linspace

mindspore.ops.linspace(start, stop, num)[源代码]

返回一个在区间 start 和 stop （包括 start 和 stop ）内均匀分布的，包含 num 个值的1维Tensor。

\[\begin{split}\begin{aligned} &step = (stop - start)/(num - 1)\\ &output = [start, start+step, start+2*step, ... , stop] \end{aligned}\end{split}\]

参数：

  • start (Tensor) - 0维Tensor，数据类型必须为float32。区间的起始值。

  • stop (Tensor) - 0维Tensor，数据类型必须为float32。区间的末尾值。

  • num (int) - 间隔中的包含的数值数量，包括区间端点。必须为正数。

返回：

Tensor，具有与 start 相同的dtype，shape为 \((num)\) 。

异常：

  • TypeError - start 或 stop 不是Tensor。

  • TypeError - start 或 stop 的数据类型不是float32。

  • ValueError - start 或 stop 的维度不是0。

  • TypeError - num 不是int类型。

  • ValueError - num 不是正数。

支持平台：

Ascend GPU CPU

样例：

>>> start = Tensor(1, mindspore.float32)
>>> stop = Tensor(10, mindspore.float32)
>>> num = 5
>>> output = ops.linspace(start, stop, num)
>>> print(output)
[ 1.    3.25  5.5   7.75 10.  ]