参考资料

pytorch训练中的一些优化手段 https://blog.csdn.net/deephub/article/details/139622843
pytorch中python和底层C++ tensor对象关系 https://pytorch.org/cppdocs/notes/tensor_basics.html#:~:text=The ATen tensor library backing PyTorch is a,API is generic and does not include templates.
pytorch训练中提到的一些优化手段 https://www.jiqizhixin.com/articles/2021-01-17-2