mindspore.ops.LinSpace

查看源文件
class mindspore.ops.LinSpace[源代码]

创建一个一维Tensor,其包含steps个元素,这些元素在区间[start, end]内均匀分布。

step=(endstart)/(steps1)output=[start,start+step,start+2step,...,end]
输入:
  • start (Tensor) - 区间的起始值。

  • stop (Tensor) - 区间的末尾值。

  • num (Union[int, Tensor]) - 元素数量。

输出:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> start = mindspore.tensor(3, mindspore.float32)
>>> stop = mindspore.tensor(10, mindspore.float32)
>>> num = 5
>>> output = mindspore.ops.LinSpace()(start, stop, num)
>>> print(output)
[ 3.    4.75  6.5   8.25 10.  ]
>>>
>>> start = mindspore.tensor(-10, mindspore.float32)
>>> stop = mindspore.tensor(10, mindspore.float32)
>>> num = 5
>>> output = mindspore.ops.LinSpace()(start, stop, num)
>>> print(output)
[-10.  -5.   0.   5.  10.]
>>>
>>> start = mindspore.tensor(-10, mindspore.float32)
>>> stop = mindspore.tensor(10, mindspore.float32)
>>> num = 1
>>> output = mindspore.ops.LinSpace()(start, stop, num)
>>> print(output)
[-10.]