If I case the our_dataset to float32 and labels to int64, then the error changes to:
multi-target not supported at /data/users/soumith/miniconda2/conda-bld/pytorch-0.1.7_1485444530918/work/torch/lib/THNN/generic/ClassNLLCriterion.c:20
but the shape of the labels is (10000,).
Hmm, this is becoming weird.