像使用numpy的arrray一样,使用pytorch的tensor
import torch
# numpy array -> tensor
pt_tensor1 = torch.Tensor(numpy_array)
- Variable
Variable是对Tensor的封装,操作与tensor基本一致
像使用numpy的arrray一样,使用pytorch的tensor
import torch
# numpy array -> tensor
pt_tensor1 = torch.Tensor(numpy_array)