这是来自自定义 Keras 回调 casted=K.cast((yPred(, K.tf.int32(
我绝对需要将 yPred(一个张量(转换为 int32 类型(该强制转换为张量内容,我知道(
尽管如此,K.cast只允许转换为浮动。
如何解决问题?
这是你这样做的方式:
casted = K.cast(yPred,"int32")
如果你的后端是 Tensorflow,则转换为 int 将起作用。
Keras.backend
接口必须是其所有可能的后端中最低的公分母。cast
CNTK(另一个可能的cast
后端(不支持 int 到 int。
如果你总是计划使用 TensorFlow 作为 Keras 的后端,那么转换为 int 将始终有效。
https://www.tensorflow.org/api_docs/python/tf/dtypes/cast
https://cntk.ai/pythondocs/cntk.ops.html