Function burn_tensor::module::conv2d

source ·

pub fn conv2d<B>(
    x: &Tensor<B, 4>,
    weight: &Tensor<B, 4>,
    bias: Option<&Tensor<B, 1>>,
    stride: [usize; 2],
    padding: [usize; 2]
) -> Tensor<B, 4>where
    B: Backend,

Expand description

Applies a 2D convolution.