You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
9 lines
206 B
9 lines
206 B
1 month ago
|
from tinygrad.tensor import Tensor
|
||
|
|
from tinygrad.nn import Embedding
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
vocab_size = 50257
|
||
|
|
dim = 128
|
||
|
|
test = Embedding(vocab_size, dim)
|
||
|
|
ret = test(Tensor([[1,2,3]])).numpy()
|