nparray和tensor的相互转化

1.array转化为tensor

x = torch.from_numpy(lena_img)

2.tensor转化为array

x=x.numpy()

猜你喜欢

转载自blog.csdn.net/huatianxue/article/details/108298113