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

Question

这来自于自定义Keras回调cast = K.cast（（yPred），K.tf.int32）

我绝对需要将类型为Tensor的yPred强制转换为int32类型（强制转换应用于Tensor内容，我知道）

尽管如此，K.cast只允许转换为浮点型。

我该如何解决这个问题？

Answer 1

这是您的操作方式：casted = K.cast(yPred,"int32")