当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


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