Files
onepilot/tinygrad_repo/test/external/external_test_mnist_data_select.py
Vehicle Researcher 6fe5d4054d openpilot v0.11.0 release
date: 2026-03-13T21:04:16
master commit: 41994b1d27423e8a10008019a962417bf0f5f974
2026-03-13 21:04:16 -07:00

9 lines
263 B
Python

#!/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()