1use candle::Tensor; 2 3pub struct Dataset { 4 pub train_images: Tensor, 5 pub train_labels: Tensor, 6 pub test_images: Tensor, 7 pub test_labels: Tensor, 8 pub labels: usize, 9} 10 11pub mod cifar; 12pub mod mnist;