tf.inside_function

Указывает, выполняется ли вызывающий код внутри tf.function.

tf.inside_function() -> bool

Возвращаемое значение
Булево значение, True, если вызывающий код выполняется внутри `tf.function`, а не по-спешному.

tf.inside_function()
False
@tf.function
def f():
  print(tf.inside_function())
f()
True

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/inside_function