diff --git a/test/external/external_test_mnist_data_select.py b/test/external/external_test_mnist_data_select.py new file mode 100644 index 0000000000..d4fd6c0a57 --- /dev/null +++ b/test/external/external_test_mnist_data_select.py @@ -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()