Skip to main content

conv2d

burn_tensor::module

Function conv2d 

Source 
pub fn conv2d(
    x: Tensor<4>,
    weight: Tensor<4>,
    bias: Option<Tensor<1>>,
    options: impl Into<PaddedConvOptions<2>>,
) -> Tensor<4>
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.