mindspore.mint.linspace

查看源文件
mindspore.mint.linspace(start, end, steps, *, dtype=None)[源代码]

创建一个steps个元素的,在[start, end]区间内均匀分布的一维tensor。

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

警告

Atlas训练系列产品暂不支持int16数据类型。

参数:
  • start (Union[float, int]) - 区间的起始值。

  • end (Union[float, int]) - 区间的末尾值。

  • steps (int) - 元素数量。

关键字参数:
返回:

Tensor

支持平台:

Ascend

样例:

>>> import mindspore
>>> output = mindspore.mint.linspace(3, 10, 5)
>>> print(output)
[ 3.    4.75  6.5   8.25 10.  ]
>>> output = mindspore.mint.linspace(-10, 10, 5)
>>> print(output)
[-10.  -5.   0.   5.  10.]
>>> output = mindspore.mint.linspace(-10, 10, 1)
>>> print(output)
[-10.]