When looking at the documentation for torch.det for Pytorch version 1.1.0 they don’t have this feature as an example, so it’s most likely what you mentioned that it’s a feature that was made available at later releases. It seems that they introduced this in 1.2.0 (this is when the docs have an example on this).