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: tensor.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.