package mydl.dataset;

import java.util.ArrayList;

import mydl.tensor.Tensor;

public class Test {
    public static void main(String[] args) {
        ArrayList<Tensor> train_image = MNIST.readTrainImage2D();
        System.out.println(train_image.size());
        ArrayList<Tensor> train_label = MNIST.readTrainLabel();
        System.out.println(train_label.size());
        ArrayList<Tensor> test_image = MNIST.readTestImage2D();
        System.out.println(test_image.size());
        ArrayList<Tensor> test_label = MNIST.readTestLabel();
        System.out.println(test_label.size());
    }
}

 

Output:

60000
60000
10000
10000