摘要: 因为一些原因,需要把Pytorch框架下的一个模块转换为numpy的形式,主要涉及到tensor到array的对应,现记录如下: 关于np.nonzero的理解,链接描述清楚,可参考:https://blog.csdn.net/u013698770/article/details/54632047, 阅读全文
posted @ 2021-04-08 13:01 vv_869 阅读(370) 评论(0) 推荐(0)