- 將Numpy矩陣轉換爲Tensor張量
sub_ts = torch.from_numpy(sub_img)
- 將Tensor張量轉化爲Numpy矩陣
sub_np1 = sub_ts.numpy()
- 將Tensor轉換爲Variable
sub_va = Variable(sub_ts)
- 將Variable轉換爲Tensor
sub_np2 = sub_va.data
pytorch入門教程之(二)-Variable、Tensor、Numpy之間的轉化
發表評論
所有評論
還沒有人評論，想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章