So what im not sure is that is the torch.min causing this error. Or is it some prior operation on the state (as pasted above) and the error is just being raised by that call – is there a way to identify this?