Pytorch dataset transform.