221 Star 941 Fork 693

GVPMindSpore/mindscience

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
sponge.function.gather_value.rst 534 Bytes
一键复制 编辑 原始数据 按行查看 历史
Li Qingguo 提交于 2023-10-11 10:29 +08:00 . fix doc

sponge.function.gather_value

.. py:function:: sponge.function.gather_value(tensor: Tensor, index: Tensor)

    根据指标从张量的最后一根轴收集值。

    参数:
        - **tensor** (Tensor) - 输入张量,shape为 :math:`(B, X)` 。其中 :math:`B` 为batch size, :math:`X` 为输入张量第二维的任意大小。
        - **index** (Tensor) - 索引,shape为 :math:`(B, ...,)` 。

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

搜索帮助