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.