Unfortunately not, but this is something that we have always thought was a good idea, just haven’t gotten around to implementing. See https://github.com/pytorch/pytorch/issues/5013 17 (although I see you have seen it already!)