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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。