因为一些原因,需要把Pytorch框架下的一个模块转换为numpy的形式,主要涉及到tensor到array的对应,现记录如下:
关于np.nonzero的理解,链接描述清楚,可参考:https://blog.csdn.net/u013698770/article/details/54632047,注意返回值即可。
先敲一点,后续补充。。。
忙碌且充实。
因为一些原因,需要把Pytorch框架下的一个模块转换为numpy的形式,主要涉及到tensor到array的对应,现记录如下:
关于np.nonzero的理解,链接描述清楚,可参考:https://blog.csdn.net/u013698770/article/details/54632047,注意返回值即可。
先敲一点,后续补充。。。
忙碌且充实。