pub fn conv2d( in_channels: usize, out_channels: usize, kernel_size: usize, cfg: Conv2dConfig, vs: VarBuilder<'_>, ) -> Result<Conv2d>