1. 示例代碼
import torch
import numpy as np
if __name__=="__main__":
np_data=np.arange(6).reshape(2,3)
# np_data不可能有from_numpy這樣方法,所以由torch包提供
torch_data=torch.from_numpy(np_data)
# 而numpy()完全可以由tensor數(shù)據(jù)類型提供
torch2np=torch_data.numpy()
print(np_data,'\n',torch_data,'\n',torch2np)
2. from_numpy
np_data不可能有from_numpy這樣方法,所以由torch包提供
3. numpy()
而numpy()完全可以由tensor數(shù)據(jù)類型提供
4. FloatTensor(list)
直接將list傳入構(gòu)造函數(shù)FloatTensor中