Short name describing what triggered the graph break
torch.while_loop: unsupported cond_fn return type
Values or code snippet captured at the break point
str(cond_r)
Explanation of why the graph break was triggered
Expected cond_fn to return a scalar tensor or a bool but got {cond_r_meta.shape}.
Hints on how to resolve the graph break