Hi ptrblck,
Thanks for the solution. It took me ages to find this page and implement the solution. This is not obvious and totally different from other common AI/ML frameworks. Is there a webpage/document that someone can access this information in the first place?
It will be great if such an examples are added to the PyTorch tutorials/examples.
Siamak