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

Question

这是来自一个自定义的Keras回调 铸造=K.铸造（（yPred），K。tf.int32型)在

我绝对需要将yPred转换为int32类型（转换应用于张量内容，我知道）

不过，K.cast只允许转换为float。在

我怎样才能解决这个问题？在

Answer 1

您可以这样做：

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

Answer 2

如果您的后端是Tensorflow，那么转换为int就可以了。在

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

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

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

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