Add openpilot tests
This commit is contained in:
8
tinygrad_repo/test/external/external_test_embedding.py
vendored
Normal file
8
tinygrad_repo/test/external/external_test_embedding.py
vendored
Normal file
@@ -0,0 +1,8 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user