pytorch中tensor與numpy的相互轉換

Tensor轉NumPy

使用numpy（）函數進行轉換

例子

 

 

NumPy數組轉Tensor

使用torch.from_numpy（）函數

例子

 

 注意事項

這兩個函數所產⽣的的 Tensor 和NumPy中的數組共享相同的內存（所以他們之間的轉換很快），改變其中⼀個時另⼀個也會改變！！！

NumPy中的array轉換成 Tensor 的⽅法還有就是 torch.tensor() , 需要注意的 是，此⽅法總是會進⾏數據拷⻉（就會消耗更多的時間和空間），所以返回的 Tensor 和原來的數 據不再共享內存