Function burn::tensor::module::embedding

pub fn embedding<B>(
    weights: Tensor<B, 2>,
    indices: Tensor<B, 2, Int>,
) -> Tensor<B, 3>
where
    B: Backend,

Applies the embedding module.