Function burn::tensor::module::conv2d

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

Applies a 2D convolution.