这来自自定义Keras回调 casted = K.cast((yPred),K.tf.int32)
我绝对需要将类型为Tensor的yPred强制转换为int32类型(强制转换应用于Tensor内容,我知道)
仍然,K.cast仅允许转换为浮点型。
我该如何解决问题?
答案 0 :(得分:3)
这是您的操作方式:
casted = K.cast(yPred,"int32")
答案 1 :(得分:0)
如果您的后端是Tensorflow,则可以将数据转换为整数。
在所有可能的后端中,Keras.backend
接口必须是最低的公分母。 TensorFlow的cast
支持int。 CNTK,另一个可能的后端,不支持cast
进行整型。
如果您始终打算将TensorFlow用作Keras的后端,则始终可以强制转换为int。