mlx.nn.MaxPool1d

目录

mlx.nn.MaxPool1d#

class MaxPool1d(kernel_size: int | Tuple[int], stride: int | Tuple[int] | None = None, padding: int | Tuple[int] = 0)#

应用一维最大池化。

通过选取一个大小为 kernel_size 的滑动窗口的最大值，并按照滑动步长 stride 进行，对输入进行空间下采样。

参数:

  • kernel_size (int 或 tuple(int)) – 池化窗口核的大小。

  • stride (int 或 tuple(int), 可选) – 池化窗口的步长。默认值: kernel_size。

  • padding (int 或 tuple(int), 可选) – 在输入上应用多少负无穷大填充。填充量应用于空间轴的两侧。默认值: 0。

示例

>>> import mlx.core as mx
>>> import mlx.nn.layers as nn
>>> x = mx.random.normal(shape=(4, 16, 5))
>>> pool = nn.MaxPool1d(kernel_size=2, stride=2)
>>> pool(x)

方法

目录