pub fn conv2d( in_channels: usize, out_channels: usize, kernel_size: usize, cfg: Conv2dConfig, vb: VarBuilderArgs<'_, ShardedSafeTensors<'_>>, ) -> Result<Conv2d, Error>