I have the same question. It seems there are some testing utils in the repo https://github.com/pytorch/pytorch/blob/master/test/common.py 279
but I don’t think they’re exposed in the distribution package, are they?