tfm.vision.layers.SpatialAveragePool3D

Creates a global average pooling layer pooling across spatial dimentions.

tfm.vision.layers.SpatialAveragePool3D(
    keepdims: bool = False, **kwargs
)

keepdims A bool. If True, keep the averaged dimensions.
**kwargs Additional keyword arguments to be passed to this layer.

Methods

call

View source

call(
    inputs, states=None, output_states: bool = False
)

Calls the layer with the given inputs.