2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.mint.func_linspace.rst 1.21 KB
一键复制 编辑 原始数据 按行查看 历史

mindspore.mint.linspace

.. py:function:: mindspore.mint.linspace(start, end, steps, *, dtype=None)

    返回一个在区间 `start` 和 `end` (包括 `start` 和 `end` )内均匀分布的,包含 `steps` 个值的一维Tensor。

    .. warning::
        Atlas训练系列产品暂不支持int16数据类型。

    .. math::
        \begin{aligned}
        &step = (end - start)/(steps - 1)\\
        &output = [start, start+step, start+2*step, ... , end]
        \end{aligned}

    参数:
        - **start** (Union[float, int]) - 区间的起始值。可以为int或float。
        - **end** (Union[float, int]) - 区间的末尾值。可以为int或float。
        - **steps** (int) - 间隔中的包含的数值数量,包括区间端点。必须为正整数。

    关键字参数:
        - **dtype** (:class:`mindspore.dtype`, 可选) - 期望输出Tensor的类型。默认值: ``None`` ,则输出类型为float32。

    返回:
        Tensor,具有与 `start` 相同的dtype,shape为 :math:`(steps)` ,数据类型由 `dtype` 指定。

    异常:
        - **TypeError** - `start` 或 `end` 的数据类型不支持。
        - **ValueError** - `steps` 不是正整数。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助