tf.estimator.export.TensorServingInputReceiver
| Просмотреть исходный код на GitHub |
Тип возвращаемого значения для serving_input_receiver_fn.
tf.estimator.export.TensorServingInputReceiver(
features, receiver_tensors, receiver_tensors_alternatives=None
)
Это используется с моделями, которые ожидают одного Tensor или SparseTensor в качестве входного признака, а не словаря признаков.
Обычный ServingInputReceiver всегда возвращает словарь признаков, даже если он содержит только одну запись, и поэтому может использоваться только с моделями, принимающими такой словарь. Для моделей, принимающих только один исходный признак, serving_input_receiver_fn , предоставленный для Estimator.export_saved_model(), должен возвращать этот TensorServingInputReceiver вместо него. См.: https://github.com/tensorflow/tensorflow/issues/11674
Обратите внимание, что аргументы receiver_tensors и receiver_tensor_alternatives будут автоматически преобразованы в представление словаря в любом случае, потому что формат SavedModel требует, чтобы каждый входной Tensor имел имя (предоставленное ключом словаря).
| Атрибуты | |
|---|---|
features | Один Tensor или SparseTensor, представляющий признак, который будет передан модели. |
receiver_tensors | Tensor, SparseTensor, или словарь из строки в Tensor или SparseTensor, определяющий входные узлы, где этот приемник ожидает по умолчанию ввода. Как правило, это один placeholder, ожидающий сериализованные tf.Example протоколы. |
receiver_tensors_alternatives | словарь из строки в дополнительные группы приемных тензоров, каждая из которых может быть Tensor, SparseTensor, или словарь из строки в Tensor или SparseTensor. Эти именованные альтернативные тензоры приемника создают дополнительные сигнатуры для обслуживания, которые могут использоваться для подачи входных данных в различных точках подграфа приемника ввода. Типичное использование – это возможность подачи исходных признаков Tensor потоком вниз от операции tf.parse_example(). По умолчанию равно None. |
© 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/estimator/export/TensorServingInputReceiver
