tf.OptionalSpec

Спецификация типа для tf.experimental.Optional.

Наследуется от: TypeSpec

tf.OptionalSpec(
    element_spec
)

Например, tf.OptionalSpec можно использовать для определения tf.function, который принимает tf.experimental.Optional в качестве входного аргумента:

@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)

Методы

from_value

Просмотреть исходный код

@staticmethod
from_value(
    value
)

is_compatible_with

Просмотреть исходный код

is_compatible_with(
    spec_or_value
)

Возвращает true, если spec_or_value совместим с этим TypeSpec.

most_specific_compatible_type

Просмотреть исходный код

most_specific_compatible_type(
    other
)

Возвращает самую специфическую TypeSpec, совместимую с self и other.

__eq__

Просмотреть исходный код

__eq__(
    other
)

Возвращает self==value.

__ne__

Просмотреть исходный код

__ne__(
    other
)

Возвращает self!=value.