• Performs an N-D pooling operation

    Type Parameters

    Parameters

    • input: TensorLike | T

      The input tensor, of rank 4 or rank 3 of shape [batch, height, width, inChannels]. If rank 3, batch of 1 is assumed.

    • windowShape: number | [number, number]

      The filter size: [filterHeight, filterWidth]. If filterSize is a single number, then filterHeight == filterWidth.

    • poolingType: "avg" | "max"

      The type of pooling, either 'max' or 'avg'.

    • pad: number | "same" | "valid" | ExplicitPadding

      The type of padding algorithm:

    • 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.

    Returns T

    Doc

Generated using TypeDoc