参考:https://zhuanlan.zhihu.com/p/102904450 与嘉神的 https://blog.csdn.net/ssjdoudou/article/details/105566465 结束会使用自己的demo讨论
踩坑1. 使用torch进行索引时,但是数据为numpy数组,则会出现报错
too many indices for tensor of dimension 1
这里需要把格式统一,也就是~tensor就是tensor,numpy就是numpy,保持一致的风格!!!。