Function burn_tensor::module::embedding

pub fn embedding<B>(
    weights: &Tensor<B, 2>,
    indexes: &Tensor<B::IntegerBackend, 2>
) -> Tensor<B, 3>where
    B: Backend,