TracerWarning: Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future.
Some results have been hidden because they may be inaccessible to you
Show inaccessible results
Feedback