Function burn_tensor::module::embedding

source · 
pub fn embedding<B>(
    weights: Tensor<B, 2>,
    indexes: Tensor<B, 2, Int>
) -> Tensor<B, 3>where
    B: Backend,
Expand description

Applies the embedding module.