mirror of
https://github.com/firestar5683/StarPilot.git
synced 2026-09-03 22:53:44 +08:00
github-actions[bot]
eefc084302
version: sunnypilot v2025.003.000 (dev) date: 2025-12-18T05:48:43 master commit: 16c052af0835f049a67b80251d8cc1114fc08bb4
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()
|