Mnist dataset for Torch