import { Callout } from "nextra-theme-docs" import { Tab, Tabs } from "nextra-theme-docs"
Average pooling layer that averages over the penultimate dimension of an input tensor.
AvgPool()
(x: jaxtyping.Float[Tensor, '... n d']) -> jaxtyping.Float[Tensor, '... d']