Skip to main content

conv2d

Function conv2d 

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

Applies a 2D convolution.

Supports symmetric and asymmetric padding through ConvOptions. The deprecated PaddedConvOptions is also accepted for compatibility.