• Computes a 3D convolution over the input x.

    Type Parameters

    Parameters

    • x: TensorLike | T

      The input tensor, of rank 5 or rank 4, of shape [batch, depth, height, width, channels]. If rank 4, batch of 1 is assumed.

    • filter: TensorLike | Tensor5D

      The filter, rank 5, of shape [filterDepth, filterHeight, filterWidth, inChannels, outChannels]. inChannels must match between input and filter.

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

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

    • pad: "same" | "valid"

      The type of padding algorithm.

    • Optional dataFormat: "NDHWC" | "NCDHW"
    • Optional dilations: number | [number, number, number]

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

    Returns T

    Doc

Generated using TypeDoc