tf.io.encode_proto

Операция сериализует сообщения protobuf, переданные во входных тензорах.

tf.io.encode_proto(
    sizes: Annotated[Any, _atypes.Int32],
    values,
    field_names,
    message_type: str,
    descriptor_source: str = 'local://',
    name=None
) -> Annotated[Any, _atypes.String]

Типы тензоров в values должны соответствовать схеме полей, указанных в field_names. Все тензоры в values должны иметь общий префикс формы — batch_shape.

Тензор sizes задает количество повторений для каждого поля. Количество повторений (последнее измерение) каждого тензора в values должно быть не меньше соответствующего количества повторений в sizes.

Чтобы задать контекст для имен полей, необходимо указать имя message_type. Дескриптор фактического сообщения можно найти либо в подключенном пуле дескрипторов, либо в файле, указанном вызывающим кодом с помощью атрибута descriptor_source.

В большинстве случаев сопоставление типов полей Proto с типами данных TensorFlow очевидно. Однако есть несколько особых случаев:

• Поле proto, содержащее подсообщение или группу, можно преобразовать только в DT_STRING (сериализованное подсообщение). Это сделано для упрощения API. Полученную строку можно использовать в качестве входных данных для другого экземпляра операции decode_proto.

• TensorFlow не поддерживает беззнаковые целые числа. В операциях типы uint64 представлены как DT_INT64 с тем же битовым шаблоном в дополнительном коде (самый очевидный способ). Значения uint32 можно точно представить, указав тип DT_INT64 или используя дополнительный код, если вызывающий код указывает DT_INT32 в атрибуте output_types.

Атрибут descriptor_source выбирает источник дескрипторов протокола, используемый для поиска message_type. Это может быть:

• Пустая строка или "local://"; в этом случае дескрипторы протокола создаются для определений proto на C++ (не Python), подключенных к исполняемому файлу.

• Файл; в этом случае дескрипторы протокола создаются из файла, который, как предполагается, содержит FileDescriptorSet, сериализованный в виде строки. ПРИМЕЧАНИЕ. Файл descriptor_source можно создать с помощью параметров --descriptor_set_out и --include_imports компилятора протоколов protoc.

• "bytes://"; в этом случае дескрипторы протокола создаются из <bytes>, который, как предполагается, представляет собой FileDescriptorSet, сериализованный в виде строки.