В ролях KERAS Тензор к K.tf.int32

Question

1

Это из пользовательского обратного вызова Keras casted = K.cast((yPred), K.tf.int32)

Мне абсолютно нужно использовать yPred, который является тензором, для типа int32 (Приведение применяется к содержимому Tensor, я это знаю)

Тем не менее, K.cast допускает только преобразование в float.

Как я могу решить проблему?

Daniel Möller 22 июнь 2018, в 20:00

Теги:

tensorflow

keras

tensor

casting

1 ответ

Daniel Möller · Accepted Answer · 2018-06-22T16-55-00.000Z

Вот как вы это делаете:

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

Как вы сказали, даже если бы я был скептиком, это работает. tenorflow.org/api_docs/python/tf/keras/backend/cast не согласен, что меня очень удивляет .... но я дал ему попробовать объявление: Tensor ("loss_2 / density_42_loss / Cast: 0", форма = (?, 1), dtype = int32). , , Спасибо!
Ответ правильный, но мои проблемы еще не закончены. Здесь мы идем: stackoverflow.com/questions/50994041/… . Это может быть интересно тем, кто оказался в этом вопросе