tf.TypeSpec

Определяет тип значения TensorFlow.

Унаследовано от: TraceType

A tf.TypeSpec предоставляет метаданные, описывающие объект, принимаемый или возвращаемый API TensorFlow. Конкретные подклассы, такие как tf.TensorSpec и tf.RaggedTensorSpec, используются для описания различных типов значений.

Например, аргумент input_signature метода tf.function принимает список (или вложенную структуру) TypeSpec.

Создание новых подклассов TypeSpec (вне ядра TensorFlow) в настоящее время не поддерживается. В частности, мы можем внести несовместимые изменения в закрытые методы и свойства, определенные в этом базовом классе.

Пример:

spec = tf.TensorSpec(shape=[None, None], dtype=tf.int32)
@tf.function(input_signature=[spec])
def double(x):
  return x * 2
double(tf.constant([[1, 2], [3, 4]]))
<tf.Tensor: shape=(2, 2), dtype=int32,
    numpy=array([[2, 4], [6, 8]], dtype=int32)>

Методы

experimental_as_proto

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

experimental_as_proto() -> struct_pb2.TypeSpecProto

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

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

experimental_from_proto

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

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

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

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

experimental_type_proto

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

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

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

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

is_compatible_with

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

is_compatible_with(
    spec_or_value
)

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

Предпочтительно использовать "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.