2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_cast.rst 757 Bytes
一键复制 编辑 原始数据 按行查看 历史
wujueying 提交于 2024-08-07 20:04 . ops.cast add docs

mindspore.ops.cast

.. py:function:: mindspore.ops.cast(input, dtype)

    转换输入Tensor的数据类型。

    .. note::
        将复数转换为bool类型的时候,不考虑复数的虚部,只要实部不为零,返回True,否则返回False。

    参数:
        - **input** (Union[Tensor, Number]) - 输入要进行数据类型转换的Tensor,其shape为 :math:`(x_1, x_2, ..., x_R)` 。
        - **dtype** (dtype.Number) - 指定转换的数据类型。仅支持常量值。

    返回:
        Tensor,其shape与 `input` 相同,即 :math:`(x_1, x_2, ..., x_R)` 。

    异常:
        - **TypeError** - `input` 既不是Tensor也不是数值型。
        - **TypeError** - `dtype` 不是数值型。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助