Skip to main content

conv2d

Function conv2d 

Source
pub fn conv2d<B>(
    x: Tensor<B, 4>,
    weight: Tensor<B, 4>,
    bias: Option<Tensor<B, 1>>,
    options: impl Into<PaddedConvOptions<2>>,
) -> Tensor<B, 4>
where B: Backend,
Expand description

Applies a 2D convolution.

Accepts ConvOptions for symmetric padding, or PaddedConvOptions for asymmetric padding. When asymmetric padding is specified, an explicit pad operation is applied before the convolution backend op.