pytorch numpy 转换成 tensor ——》 torch.from_numpy()

sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型

posted @ 2022-03-20 21:36  bH1pJ  阅读(52)  评论(0)  编辑  收藏  举报