Embedding table lookup (Geron Ch 16, morie.fn gremb)

Usage

morie_geron_embedding_lookup(ids, E)

Arguments

ids: 0-based integer token ids.
E: (V, d) embedding table.

Value

List with embeddings, vocab_size, dim, n_unique, n_parameters, ids.

Examples

E <- matrix(seq(0.1, 1.2, by = 0.1), 4, 3)
morie_geron_embedding_lookup(ids = c(1L, 3L, 2L), E = E)
#> $embeddings
#>      [,1] [,2] [,3]
#> [1,]  0.2  0.6  1.0
#> [2,]  0.4  0.8  1.2
#> [3,]  0.3  0.7  1.1
#> 
#> $vocab_size
#> [1] 4
#> 
#> $dim
#> [1] 3
#> 
#> $n_unique
#> [1] 3
#> 
#> $n_parameters
#> [1] 12
#> 
#> $ids
#> [1] 1 3 2
#> 
#> $estimate
#>      [,1] [,2] [,3]
#> [1,]  0.2  0.6  1.0
#> [2,]  0.4  0.8  1.2
#> [3,]  0.3  0.7  1.1
#> 
#> $n
#> [1] 3
#>