tf.io.encode_proto
Операция сериализует сообщения protobuf, предоставленные в тензорах входных данных.
tf.io.encode_proto(
sizes,
values,
field_names,
message_type,
descriptor_source='local://',
name=None
)
Типы тензоров в values должны соответствовать схеме полей, указанных в field_names. Все тензоры в values должны иметь общий префикс формы, batch_shape.
Тензор sizes указывает количество повторений для каждого поля. Количество повторений (последнее измерение) каждого тензора в values должно быть больше или равно соответствующему количеству повторений в sizes.
Необходимо указать имя message_type, чтобы предоставить контекст для имен полей. Фактический описатель сообщения можно найти либо в связанном пуле описателей, либо в имени файла, предоставленном вызывающей стороной с помощью атрибута descriptor_source.
В основном, отображение между типами полей прото и типами данных TensorFlow прямое. Однако есть несколько особых случаев:
Поле прото, содержащее подсообщение или группу, может быть преобразовано только в
DT_STRING(сериализованное подсообщение). Это сделано для уменьшения сложности API. Полученную строку можно использовать в качестве входных данных для другого экземпляра операции decode_proto.В TensorFlow отсутствует поддержка беззнаковых целых чисел. Операции представляют типы uint64 как
DT_INT64с тем же битовым шаблоном со взаимным дополнением (очевидный способ). Беззнаковые значения int32 можно точно представить, указав типDT_INT64, или используя взаимное дополнение, если вызывающая сторона указываетDT_INT32в атрибутеoutput_types.
Атрибут descriptor_source выбирает источник протокольных описателей для поиска message_type. Это может быть:
Пустая строка или «local://», в этом случае протокольные описатели создаются для определений прото в C++ (а не Python), связанных с двоичным файлом.
Файл, в этом случае протокольные описатели создаются из файла, который, как ожидается, содержит
FileDescriptorSet, сериализованный в виде строки. ПРИМЕЧАНИЕ: Вы можете создать файлdescriptor_sourceс помощью параметров--descriptor_set_outи--include_importsдля компилятора протоколовprotoc.«bytes://
», в этом случае протокольные описатели создаются из <bytes>, который, как ожидается, представляет собойFileDescriptorSet, сериализованный в виде строки.
| Аргументы | |
|---|---|
sizes | A Tensor типа int32. Тензор int32 с формой [batch_shape, len(field_names)]. |
values | Список объектов Tensor. Список тензоров, содержащих значения для соответствующего поля. |
field_names | Список strings . Список строк, содержащих имена полей прото. |
message_type | string . Имя типа сообщения прото для декодирования. |
descriptor_source | Необязательный string . По умолчанию "local://" . |
name | Имя операции (необязательно). |
| Возвращаемое значение | |
|---|---|
A Tensor типа string . |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/io/encode_proto