Pytorch Dataset Basic Form Vs Special Form