cerebras.modelzoo.fc_mnist.pytorch.prepare_data#