tf.TensorSpec

Описание типа tf.Tensor.

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

tf.TensorSpec(
    shape,
    dtype=tf.dtypes.float32,
    name=None
)

Используется в блокнотах

t = tf.constant([[1,2,3],[4,5,6]])
tf.TensorSpec.from_tensor(t)
TensorSpec(shape=(2, 3), dtype=tf.int32, name=None)

Содержит метаданные для описания природы tf.Tensor объектов, принимаемых или возвращаемых некоторыми API TensorFlow.

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

@tf.function(input_signature=[tf.TensorSpec([1, None])])
def constrained_foo(t):
  print("tracing...")
  return t

Теперь tf.function может предположить, что t всегда имеет тип tf.TensorSpec([1, None]), что позволит избежать повторного прослеживания, а также обеспечить ограничение типа входных данных.

В результате следующий вызов с тензором типа tf.TensorSpec([1, 2]) запускает прослеживание и завершается успешно:

>>> constrained_foo(tf.constant([[1., 2]])).numpy()
tracing...
array([[1., 2.]], dtype=float32)

Следующий последующий вызов с тензором типа tf.TensorSpec([1, 4]) не запускает прослеживание и завершается успешно:

>>> constrained_foo(tf.constant([[1., 2, 3, 4]])).numpy()
array([[1., 2., 3., 4.], dtype=float32)

Но следующий вызов с тензором типа tf.TensorSpec([2, 2]) завершается неудачно:

>>> constrained_foo(tf.constant([[1., 2], [3, 4]])).numpy()
Traceback (most recent call last):
...
TypeError: Binding inputs to tf.function `constrained_foo` failed ...

Методы

experimental_as_proto

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

experimental_as_proto() -> struct_pb2.TensorSpecProto

Возвращает представление прото объекта TensorSpec.

experimental_from_proto

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

@classmethod
experimental_from_proto(
    proto: struct_pb2.TensorSpecProto
) -> 'TensorSpec'

Возвращает объект TensorSpec на основе сериализованного прото.

experimental_type_proto

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

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

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

from_spec

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

@classmethod
from_spec(
    spec, name=None
)

Возвращает TensorSpec с той же формой и типом данных, что и spec.

spec = tf.TensorSpec(shape=[8, 3], dtype=tf.int32, name="OriginalName")
tf.TensorSpec.from_spec(spec, "NewName")
TensorSpec(shape=(8, 3), dtype=tf.int32, name='NewName')

from_tensor

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

@classmethod
from_tensor(
    tensor, name=None
)

Возвращает TensorSpec, описывающее tensor.

tf.TensorSpec.from_tensor(tf.constant([1, 2, 3]))
TensorSpec(shape=(3,), dtype=tf.int32, name=None)

is_compatible_with

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

is_compatible_with(
    spec_or_tensor
)

Возвращает True, если spec_or_tensor совместим с этим TensorSpec.

Два тензора считаются совместимыми, если у них одинаковый тип данных и их формы совместимы (см. tf.TensorShape.is_compatible_with).

is_subtype_of

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

is_subtype_of(
    other
)

Возвращает 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
)

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

__ne__

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

__ne__(
    other
)

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