mirror of
https://github.com/MoreTore/openpilot.git
synced 2026-08-05 08:16:06 +08:00
Vehicle Researcher
ad4dbe2012
date: 2024-06-11T01:36:39 master commit: f8cb04e4a8b032b72a909f68b808a50936184bee
9 lines
206 B
Python
9 lines
206 B
Python
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()
|