1 files changed, 12 insertions, 0 deletions
diff --git a/candle-datasets/src/vision/mod.rs b/candle-datasets/src/vision/mod.rs
new file mode 100644
index 00000000..6ce743eb
--- /dev/null
+++ b/candle-datasets/src/vision/mod.rs
@@ -0,0 +1,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;