1. Torch-numpy桥
2018-06-29 本文已影响44人
FantDing
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数据类型提供
torch2np=torch_data.numpy()
print(np_data,'\n',torch_data,'\n',torch2np)
2. from_numpy
np_data不可能有from_numpy这样方法,所以由torch包提供
3. numpy()
而numpy()完全可以由tensor数据类型提供
4. FloatTensor(list)
直接将list传入构造函数FloatTensor中