How to turn a cifar10 dataset into a tensor