The input tensor, of rank 4 or rank 3 of shape
[batch, height, width, inChannels]
. If rank 3, batch of 1 is assumed.
The filter size: [filterHeight, filterWidth]
. If
filterSize
is a single number, then filterHeight == filterWidth
.
The type of pooling, either 'max' or 'avg'.
The type of padding algorithm:
same
and stride 1: output will be of same size as input,
regardless of filter size.valid
: output will be smaller than input if filter is larger
than 1x1.Optional
dilations: number | [number, number]The dilation rates: [dilationHeight, dilationWidth]
in which we sample input values across the height and width dimensions
in dilated pooling. Defaults to [1, 1]
. If dilationRate
is a single
number, then dilationHeight == dilationWidth
. If it is greater than
1, then all values of strides
must be 1.
Optional
strides: number | [number, number]The strides of the pooling: [strideHeight, strideWidth]
. If
strides
is a single number, then strideHeight == strideWidth
.
Optional
dimRoundingMode: "floor" | "round" | "ceil"A string from: 'ceil', 'round', 'floor'. If none is provided, it will default to truncate.
Generated using TypeDoc
Performs an N-D pooling operation