将 KERAS 张量转换为 K.tf.int32



这是来自自定义 Keras 回调 casted=K.cast((yPred(, K.tf.int32(

我绝对需要将 yPred(一个张量(转换为 int32 类型(该强制转换为张量内容,我知道(

尽管如此,K.cast只允许转换为浮动。

如何解决问题?

这是你这样做的方式:

casted = K.cast(yPred,"int32")

如果你的后端是 Tensorflow,则转换为 int 将起作用。

Keras.backend接口必须是其所有可能的后端中最低的公分母。castCNTK(另一个可能的cast后端(不支持 int 到 int。

如果你总是计划使用 TensorFlow 作为 Keras 的后端,那么转换为 int 将始终有效。

https://www.tensorflow.org/api_docs/python/tf/dtypes/cast

https://cntk.ai/pythondocs/cntk.ops.html

相关内容

最新更新