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