tf.experimental.BatchableExtensionType

Тип расширения, который можно объединять в пакеты и распаковывать.

Наследуется от: ExtensionType

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

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

class Vehicle(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)

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

Методы

__eq__

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

__eq__(
    other
)

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

__ne__

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

__ne__(
    other
)

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