Add flat vector read_mnist_train