Joshua Alexander

Inference tensors cannot be saved for backward. To work around you can make a clone to get a normal tensor and use it in autograd.