分类：Numpy转换

2021
07-01

解决Numpy与Pytorch彼此转换时的坑

前言​  最近使用Numpy包与Pytorch写神经网络时，经常需要两者彼此转换，故用此笔记记录码代码时踩（菜）过的坑，网上有人说：Pytorch又被称为GPU版的Numpy，二者的许多功能都有良好的一一对应。​但在使用时还是得多多注意，一个不留神就陷入到了一根烟一杯酒，一个Bug找一宿的地步。1.1、numpy——>torch​  使用torch.from_numpy()转换，需要注意，两者共享内存。例子如下... 继续阅读 >