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