github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-01-07 22:23:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

add quick external data select test

Browse Source
This commit is contained in:
George Hotz
2024-03-02 05:38:32 -08:00
parent 9a37273d36
commit 83530a585f
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/external/external_test_mnist_data_select.py vendored Normal file
View File

@@ -0,0 +1,8 @@
#!/bin/bash
from tinygrad import Tensor
from extra.datasets import fetch_mnist
if __name__ == "__main__":
X_train, Y_train, X_test, Y_test = fetch_mnist(tensors=True)
samples = Tensor.randint(512, high=X_train.shape[0])
select = X_train[samples].realize()