• Computes the 3D max pooling.

    const x = tf.tensor5d([1, 2, 3, 4, 5, 6, 7, 8], [1, 2, 2, 2, 1]);
    const result = tf.maxPool3d(x, 2, 1, 'valid');
    result.print();

    Type Parameters

    Parameters

    • x: TensorLike | T

      The input tensor, of rank 5 or rank 4 of shape [batch, depth, height, width, inChannels].

    • filterSize: number | [number, number, number]

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

    • strides: number | [number, number, number]

      The strides of the pooling: [strideDepth, strideHeight, strideWidth]. If strides is a single number, then strideDepth == strideHeight == strideWidth.

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

      The type of padding algorithm.

    • Optional dimRoundingMode: "floor" | "round" | "ceil"

      A string from: 'ceil', 'round', 'floor'. If none is provided, it will default to truncate.

    • Optional dataFormat: "NDHWC" | "NCDHW"

      An optional string from: "NDHWC", "NCDHW". Defaults to "NDHWC". Specify the data format of the input and output data. With the default format "NDHWC", the data is stored in the order of: [batch, depth, height, width, channels]. Only "NDHWC" is currently supported.

    Returns T

    Doc

Generated using TypeDoc