mindspore.ops.LinSpace

class mindspore.ops.LinSpace[源代码]

返回一个在区间 startstop (包括 startstop )内均匀分布的,包含 num 个值的一维Tensor。

更多参考详见 mindspore.ops.linspace()

输入:
  • start (Tensor) - 区间的起始值。零维Tensor,数据类型为float32。

  • stop (Tensor) - 区间的末尾值。零维Tensor,数据类型为float32。

  • num (int) - 间隔中的包含的数值数量,包括区间端点。

输出:

Tensor,与 start 的shape和数据类型相同。

支持平台:

Ascend GPU CPU

样例:

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