tf.data.IteratorSpec

Спецификация типа для tf.data.Iterator.

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

tf.data.IteratorSpec(
    element_spec
)

Например, tf.data.IteratorSpec можно использовать для определения tf.function, принимающего tf.data.Iterator в качестве входного аргумента:

@tf.function(input_signature=[tf.data.IteratorSpec(
  tf.TensorSpec(shape=(), dtype=tf.int32, name=None))])
def square(iterator):
  x = iterator.get_next()
  return x * x
dataset = tf.data.Dataset.from_tensors(5)
iterator = iter(dataset)
print(square(iterator))
tf.Tensor(25, shape=(), dtype=int32)

Методы

experimental_as_proto

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

experimental_as_proto() -> struct_pb2.TypeSpecProto

Возвращает представление прото типа TypeSpec.

НЕ переопределять для пользовательских типов, не относящихся к TensorFlow.

experimental_from_proto

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

@classmethod
experimental_from_proto(
    proto: struct_pb2.TypeSpecProto
) -> 'TypeSpec'

Возвращает экземпляр TypeSpec на основе сериализованного прото.

НЕ переопределять для пользовательских типов, не относящихся к TensorFlow.

experimental_type_proto

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

@classmethod
experimental_type_proto() -> Type[struct_pb2.TypeSpecProto]

Возвращает тип прото, связанный с сериализацией TypeSpec.

НЕ переопределять для пользовательских типов, не относящихся к TensorFlow.

from_value

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

@staticmethod
from_value(
    value
)

is_compatible_with

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

is_compatible_with(
    spec_or_value
)

Возвращает true, если spec_or_value совместим с этой спецификацией типа.

Предпочтительнее использовать "is_subtype_of" и "most_specific_common_supertype" всякий раз, когда это возможно.

is_subtype_of

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

is_subtype_of(
    other: tf.types.experimental.TraceType
) -> bool

Возвращает True, если self является подтипом other.

Реализует интерфейс tf.types.experimental.func.TraceType.

Если не переопределено подклассом, по умолчанию предполагается, что TypeSpec является ковариантным по атрибутам, реализующим TraceType, и инвариантным по остальным атрибутам, а также по структуре и типу TypeSpec.

most_specific_common_supertype

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

most_specific_common_supertype(
    others: Sequence[tf.types.experimental.TraceType]
) -> Optional['TypeSpec']

Возвращает наиболее специфичную спецификацию супертипа TypeSpec для self и others.

Реализует интерфейс tf.types.experimental.func.TraceType.

Если не переопределено подклассом, по умолчанию предполагается, что TypeSpec является ковариантным по атрибутам, реализующим TraceType, и инвариантным по остальным атрибутам, а также по структуре и типу TypeSpec.

most_specific_compatible_type

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

most_specific_compatible_type(
    other: 'TypeSpec'
) -> 'TypeSpec'

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

Устарело. Пожалуйста, используйте most_specific_common_supertype вместо этого. Не переопределяйте эту функцию.

__eq__

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

__eq__(
    other
) -> bool

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

__ne__

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

__ne__(
    other
) -> bool

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