rotta_rs 0.0.5

a Deep Learning library with rust language
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27

use crate::{ concat, Tensor };

pub struct Embedding {
    pub num_vocab: usize,
    pub hidden: usize,
    pub embedding_list: Vec<Tensor>,
}

impl Embedding {
    pub fn forward(&self, x: &Tensor) -> Tensor {
        let tokens = x.value().value;
        let mut map = vec![];

        for token in tokens {
            map.push(&self.embedding_list[token as usize]);
        }

        let mut shape = x.shape();
        shape.push(self.hidden);
        let concat = concat(map, 0);

        concat.to_shape(shape)
        // concat.node.lock().unwrap().value.shape = shape;

        // concat
    }
}