tf.experimental.ExtensionType

Базовый класс для классов TensorFlow ExtensionType.

tf.experimental.ExtensionType(
    *args, **kwargs
)

Классы TensorFlow ExtensionType — это специализированные классы Python, которые можно использовать прозрачно с TensorFlow — например, их можно использовать с операциями, такими как tf.cond или tf.while_loop, и использовать в качестве входных или выходных данных для tf.function и слоёв Keras.

Новые классы ExtensionType определяются путём создания подкласса tf.ExtensionType, содержащего аннотации типов для всех переменных экземпляра. Поддерживаются следующие аннотации типов:

Поля, помеченные typing.Mapping, будут храниться с использованием неизменяемого типа отображения.

Значения ExtensionType являются неизменяемыми — т.е., после создания вы не можете изменить или удалить какие-либо из их членов экземпляра.

Примеры

class MaskedTensor(ExtensionType):
  values: tf.Tensor
  mask: tf.Tensor

class Toy(ExtensionType):
  name: str
  price: ops.Tensor
  features: typing.Mapping[str, tf.Tensor]

class ToyStore(ExtensionType):
  name: str
  toys: typing.Tuple[Toy, ...]

Методы

__eq__

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

__eq__(
    other
)

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

__ne__

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

__ne__(
    other
)

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