TF 1 не может использовать tf.Tensor в качестве логического условия - PullRequest
0 голосов
/ 07 мая 2020

Я изучаю TF 2, но по какой-то причине мне нужно использовать TF 1 в проекте.

Однако возникла проблема, когда я использовал тензор в качестве логического условия в if. Для простой демонстрации я буду использовать следующий пример.

import tensorflow as tf
a = tf.constant(2.0)
if a :
  print('hi')

И ошибка: using a `tf.Tensor` as a Python `bool` is not allowed in Graph execution. Use Eager execution or decorate this function with @tf.function.

Я хочу, чтобы тип был float, у вас есть идеи?

...