tf.experimental.BatchableExtensionType

ExtensionType, который можно пакетировать и распаковывать.

Наследует: ExtensionType

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

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

BatchableExtensionTypes можно использовать с API, которым требуется пакетирование или распаковка, включая Keras, tf.data.Dataset и tf.map_fn. Например:

class Vehicle(tf.experimental.BatchableExtensionType):
  top_speed: tf.Tensor
  mpg: tf.Tensor
batch = Vehicle([120, 150, 80], [30, 40, 12])
tf.map_fn(lambda vehicle: vehicle.top_speed * vehicle.mpg, batch,
          fn_output_signature=tf.int32).numpy()
array([3600, 6000,  960], dtype=int32)

Эти API используют ExtensionTypeBatchEncoder для кодирования значений ExtensionType. Кодировщик по умолчанию предполагает, что значения можно объединять в пакеты, распаковывать или объединять простым объединением в пакет, распаковкой или объединением каждого вложенного поля Tensor, ExtensionType, CompositeTensor или TensorShape. Типы расширений, для которых это не так, должны переопределить __batch_encoder__ с помощью пользовательского ExtensionTypeBatchEncoder. Подробнее см. в разделе tf.experimental.ExtensionTypeBatchEncoder.

Методы

__eq__

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

__eq__(
    other
)

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

__ne__

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

__ne__(
    other
)

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