Python tf.types.experimental.GenericFunction.get_concrete_function用法及代碼示例

用法

get_concrete_function(
    *args, **kwargs
) -> tf.types.experimental.ConcreteFunction

返回專用於輸入類型的 ConcreteFunction。

args 和kwargs 指定的參數遵循正常的函數調用規則。返回的 ConcreteFunction 具有與 self 相同的位置和關鍵字參數集，但它們的類型與 args 和 kwargs 指定的類型兼容(盡管不一定相等)。

@tf.function
def f(x):
  return x
f_concrete = f.get_concrete_function(tf.constant(1.0))
f_concrete = f.get_concrete_function(x=tf.constant(1.0))

與普通調用不同，get_concrete_function 允許使用類型說明符而不是 TensorFlow 對象，因此例如 tf.Tensor 可以替換為 tf.TensorSpec 。

@tf.function
def f(x):
  return x
f_concrete = f.get_concrete_function(tf.TensorSpec([], tf.float64))

如果函數定義隻允許一種特化，args 和kwargs 可以完全省略。

@tf.function(input_signature=[tf.TensorSpec(None, tf.float32)])
def f(x):
  return x
f_concrete = f.get_concrete_function()

返回的ConcreteFunction可以正常調用：

f_concrete(tf.constant(1.0))
<tf.Tensor:shape=(), dtype=float32, numpy=1.0>
f_concrete(x=tf.constant(1.0))
<tf.Tensor:shape=(), dtype=float32, numpy=1.0>