解决pytorch 数据类型报错的问题