transforms.ToTensor()本身有维度转换功能

NoSuchKey

猜你喜欢

转载自blog.csdn.net/york1996/article/details/82711929