tf.experimental.BatchableExtensionType

Тип расширения, который можно группировать и разгруппировать.

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

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

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

BatchableExtensionType можно использовать с 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)

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

Методы

__eq__

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

__eq__(
    other
)

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

__ne__

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

__ne__(
    other
)

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