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