sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型
pytorch numpy 转换成 tensor ——》 torch.from_numpy()
Guess you like
Origin blog.csdn.net/Hodors/article/details/120846867
Ranking
sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型