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()

posted @ 2021-10-19 15:11  bH1pJ  阅读(12)  评论(0)  编辑  收藏  举报