220 Star 944 Fork 693

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
sponge.function.squeeze_first_dim.rst 403 Bytes
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2023-12-25 14:23 +08:00 . modify sponge docs format

sponge.function.squeeze_first_dim

.. py:function:: sponge.function.squeeze_first_dim(input_x: Tensor)

    返回从第一个轴(axis=0)删除大小为1的维度后的张量。

    参数:
        - **input_x** (Tensor) - 张量的shape为 :math:`(x_1, x_2, ..., x_R)`。

    返回:
        Tensor,张量的shape为 :math:`(x_2, ..., x_R)`。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/mindspore/mindscience.git
git@gitee.com:mindspore/mindscience.git
mindspore
mindscience
mindscience
master

搜索帮助