lp_pool3d

→Tensor

lp_pool3d(x: Tensor, norm_type: float, kernel_size: int | tuple[int, int, int], stride: int | tuple[int, int, int] | None = None, ceil_mode: bool = False)

source

3-D Lp-norm pooling — $\big(\sum |x|^p\big)^{1/p}$ .

Volumetric version of lp_pool2d. Interpolates between sum / average pooling (small $p$ ) and max pooling (large $p$ ) over a 3-D window.

Parameters

xTensor

Input of shape (N, C, D, H, W).

norm_typefloat

Exponent

p > 0

kernel_sizeint or (int, int, int)

Size of the pooling window per axis.

strideint or (int, int, int)= None

Window step. Defaults to kernel_size.

ceil_modebool= False

Use ceil instead of floor in the output-size formula.

Returns

Tensor

Output of shape (N, C, D_out, H_out, W_out).

Notes

Math (per window $R$ ):

y_{i,c,d,h,w} = \left( \sum_{(l, m, n) \in R} |x_{i,c,l,m,n}|^p \right)^{1/p}

Examples

>>> import lucid
>>> from lucid.nn.functional import lp_pool3d
>>> x = lucid.randn(1, 2, 8, 8, 8)
>>> y = lp_pool3d(x, norm_type=2.0, kernel_size=2)
>>> y.shape
(1, 2, 4, 4, 4)