@wandering007 there’s a pull request open to patch this: https://github.com/pytorch/pytorch/pull/7775 8 . A full fix won’t be around until pytorch supports arbitrary zero-dimensions in tensors (this should happen sometime in a few months).