mindspore.ops.LinSpace
- class mindspore.ops.LinSpace[源代码]
创建一个一维Tensor,其包含steps个元素,这些元素在区间[start, 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.]