if you do not mind this loss, you can: 1) explicitly cast your data to float32

很明显的错误提示:
这里写图片描述

一开始以为是tensor的类型不对,使用cast函数转换,发现不是,原来是numpy的问题,直接将原变量使用如下进行转换

data = data.astype(float32)

猜你喜欢

转载自blog.csdn.net/m0_37052320/article/details/79971745
今日推荐