pytorch numpy 转换成 tensor -> torch.from_numpy() . torch.Tensor to numpy
sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型
如何从torch.Tensor转换成numpy呢?

img2 = img.numpy()
sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型
如何从torch.Tensor转换成numpy呢?

img2 = img.numpy()