如何使用tensorflow 2.x 进行训练？

[panhu]

您好，我想请问下，不使用“import tensorflow.compat.v1 as tf”的情况下，使用tf 2.x进行训练，会报错：

TypeError: You are passing KerasTensor(type_spec=TensorSpec(shape=(), dtype=tf.float32, name=None), name='Placeholder:0', description="created by layer 'tf.cast_2'"), an intermediate Keras symbolic input/output, to a TF API that does not allow registering custom dispatchers, such as tf.cond, tf.function, gradient tapes, or tf.map_fn. Keras Functional model construction only supports TF API calls that do support dispatching, such as tf.math.add or tf.reshape. Other APIs cannot be called directly on symbolic Kerasinputs/outputs. You can work around this limitation by putting the operation in a custom Keras layer call and calling that layer on this symbolic input/output.

应该如何修改呢，非常感谢

[Le-Xiaohuai-speech]

使用全keras 接口试试？