pub fn embedding( input: &Tensor, weight: &Tensor, padding_idx: Option<usize>, _max_norm: Option<f32>, _norm_type: f32, _scale_grad_by_freq: bool, _sparse: bool, ) -> Tensor