add quick external data select test

2024-03-02 05:38:32 -08:00
parent 9a37273d36
commit 83530a585f
1 changed files with 8 additions and 0 deletions
@@ -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()