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

时间:2018-06-22 17:28:34

标签: python tensorflow casting keras tensor

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

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

仍然,K.cast仅允许转换为浮点型。

我该如何解决问题?

2 个答案:

答案 0 :(得分:3)

这是您的操作方式:

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

答案 1 :(得分:0)

如果您的后端是Tensorflow,则可以将数据转换为整数。

在所有可能的后端中,Keras.backend接口必须是最低的公分母。 TensorFlow的cast支持int。 CNTK,另一个可能的后端,不支持cast进行整型。

如果您始终打算将TensorFlow用作Keras的后端,则始终可以强制转换为int。

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

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