Python tf.OptionalSpec用法及代碼示例

tf.experimental.Optional 的類型規範。

繼承自：TypeSpec

用法

tf.OptionalSpec(
    element_spec
)

屬性

element_spec TypeSpec 對象的(嵌套)結構，表示可選元素的類型規範。
value_type 與此 TypeSpec 兼容的值的 Python 類型。
特別是，所有與此 TypeSpec 兼容的值都必須是此類型的實例。

例如，tf.OptionalSpec 可用於定義一個將 tf.experimental.Optional 作為輸入參數的 tf.function：

@tf.function(input_signature=[tf.OptionalSpec(
  tf.TensorSpec(shape=(), dtype=tf.int32, name=None))])
def maybe_square(optional):
  if optional.has_value():
    x = optional.get_value()
    return x * x
  return -1
optional = tf.experimental.Optional.from_value(5)
print(maybe_square(optional))
tf.Tensor(25, shape=(), dtype=int32)

相關用法

注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.OptionalSpec。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。