- 将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
sub_ts = torch.from_numpy(sub_img)
sub_np1 = sub_ts.numpy()
sub_va = Variable(sub_ts)
sub_np2 = sub_va.data