• Computes a 2D convolution over the input x.

    Type Parameters

    Parameters

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

    • filter: Tensor4D | TensorLike

      The filter, rank 4, of shape [filterHeight, filterWidth, inDepth, outDepth].

    • strides: number | [number, number]

      The strides of the convolution: [strideHeight, strideWidth].

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

      The type of padding algorithm.

    • Optional dataFormat: "NHWC" | "NCHW"
    • Optional dilations: number | [number, number]

      The dilation rates: [dilationHeight, dilationWidth] in which we sample input values across the height and width dimensions in atrous convolution. Defaults to [1, 1]. If dilations is a single number, then dilationHeight == dilationWidth. If it is greater than 1, then all values of strides must be 1.

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

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

    Returns T

    Doc