forked from mindspore-Ecosystem/mindspore
30 lines
1.3 KiB
ReStructuredText
30 lines
1.3 KiB
ReStructuredText
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` 整除。
|