mindspore.ops.LinSpace
- class mindspore.ops.LinSpace[源代码]
返回一个在区间 start 和 stop (包括 start 和 stop )内均匀分布的,包含 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 >>> 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. ]