2.4K Star 8.2K Fork 4.4K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.nn.AdaptiveAvgPool1d.rst 1.26 KB
一键复制 编辑 原始数据 按行查看 历史
DavidFFFan 提交于 2023-02-17 11:32 . refactor adaptive_max_pool1d use 2d

mindspore.nn.AdaptiveAvgPool1d

.. py:class:: mindspore.nn.AdaptiveAvgPool1d(output_size)

    在一个输入Tensor上应用1D自适应平均池化运算,可视为组成一个1D输入平面。

    通常,输入的shape为 :math:`(N_{in}, C_{in}, L_{in})` ,AdaptiveAvgPool1d在 :math:`L_{in}` 维度上计算区域平均值。
    输出的shape为 :math:`(N_{in}, C_{in}, L_{out})` ,其中, :math:`L_{out}` 为 `output_size`。

    .. note::
        :math:`L_{in}` 必须能被 `output_size` 整除。

    参数:
        - **output_size** (int) - 目标输出大小 :math:`L_{out}`。

    输入:
        - **input** (Tensor) - shape为 :math:`(N, C_{in}, L_{in})` 的Tensor,数据类型为float16或float32。

    输出:
        Tensor,其shape为 :math:`(N, C_{in}, L_{out})`,数据类型与 `input` 相同。

    异常:
        - **TypeError** - `output_size` 不是int。
        - **TypeError** - `input` 不是float16或float32。
        - **ValueError** - `output_size` 小于1。
        - **ValueError** -  `input` 的shape长度不等于3。
        - **ValueError** -  `input` 的最后一个维度小于 `output_size`。
        - **ValueError** -  `input` 的最后一个维度不能被 `output_size` 整除。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3

搜索帮助