pub fn conv2d( input: &Tensor, kernel: &Tensor, padding: [usize; 2], stride: [usize; 2], dilation: [usize; 2], groups: usize, ) -> Tensor