mindspore.ops.LinSpace

class mindspore.ops.LinSpace[源代码]

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

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

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

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

  • num (Union[int, Tensor]) - 间隔中的包含的数值数量,包括区间端点。必须为正数。当输入为Tensor时,必须是零维Tensor,数据类型为int32或int64。

输出:

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

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> 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.  ]