candle-datasets/src/vision/mod.rs


1
2
3
4
5
6
7
8
9
10
11
12

use candle::Tensor;

pub struct Dataset {
    pub train_images: Tensor,
    pub train_labels: Tensor,
    pub test_images: Tensor,
    pub test_labels: Tensor,
    pub labels: usize,
}

pub mod cifar;
pub mod mnist;