Skip to main content

embedding

Function embedding 

Source
pub fn embedding(indices: &Tensor, weight: &Tensor) -> Result<Tensor>