mindspore.ops.linspace

View Source On Gitee
mindspore.ops.linspace(start, end, steps)[source]

Returns a Tensor whose value is steps evenly spaced in the interval start and end (including start and end), and the length of the output Tensor is steps.

\[\begin{split}\begin{aligned} &step = (end - start)/(steps - 1)\\ &output = [start, start+step, start+2*step, ... , end] \end{aligned}\end{split}\]
Parameters
  • start (Union[Tensor, int, float]) – Start value of interval. The tensor data type must be float32 or float64 and with shape of 0-D.

  • end (Union[Tensor, int, float]) – Last value of interval. The tensor data type must be float32 or float64 and with shape of 0-D.

  • steps (Union[Tensor, int]) – Number of ticks in the interval, inclusive of start and end. Must be positive int number or 0D int32/int64 Tensor.

Returns

Tensor, has the same dtype as start, and the shape of \((steps)\).

Raises
  • TypeError – If start or end is not a Tensor.

  • TypeError – If dtype of start or dtype of end is not float32 or float64.

  • ValueError – If shape of start or shape of end is not 0-D.

  • TypeError – If steps is not int or 0D int32/int64 Tensor.

  • ValueError – If steps is not positive int number.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> start = Tensor(1, mindspore.float32)
>>> end = Tensor(10, mindspore.float32)
>>> steps = 5
>>> output = ops.linspace(start, end, steps)
>>> print(output)
[ 1.    3.25  5.5   7.75 10.  ]