ValueError:only one element tensors can be converted to Python scalars解决办法

ValueError:only one element tensors can be converted to Python scalars解决办法：
见
https://blog.csdn.net/qq_38703529/article/details/120216078