关于numpy和pytorch一些函数的对应和踩坑记录

因为一些原因，需要把Pytorch框架下的一个模块转换为numpy的形式，主要涉及到tensor到array的对应，现记录如下：

关于np.nonzero的理解，链接描述清楚，可参考：https://blog.csdn.net/u013698770/article/details/54632047，注意返回值即可。

 

先敲一点，后续补充。。。

 

 

 

 

 

 

 

 

 

忙碌且充实。