21 lines
851 B
ReStructuredText
21 lines
851 B
ReStructuredText
mindspore.mint.nn.AdaptiveAvgPool2d
|
||
====================================
|
||
|
||
.. py:class:: mindspore.mint.nn.AdaptiveAvgPool2d(output_size)
|
||
|
||
对由多个输入平面组成的输入信号，运用二维自适应平均池化。
|
||
|
||
对于任何输入大小，输出大小均为 :math:`H \times W`。
|
||
输出特征的数量等于输入平面的数量。
|
||
|
||
参数：
|
||
- **output_size** (Union[int, tuple[int]]) - :math:`H \times W` 形式的图像的目标输出尺寸。
|
||
可以是tuple :math:`(H, W)`，也可以是正方形图像 :math:`H \times H` 的单个 :math:`H`。
|
||
:math:`H` 和 :math:`W` 可以是 ``int`` 或 ``None``，这意味着大小将与输入相同。
|
||
|
||
输入：
|
||
- **input** (Tensor) - 输入tensor，shape为 :math:`(N, C, H, W)` 或 :math:`(C, H, W)`。
|
||
|
||
输出：
|
||
Tensor。
|