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 | Словарь, SparseTensor, или словарь из строки в Tensor или SparseTensor, определяющий входные узлы, где этот приемник ожидает по умолчанию. Как правило, это один заполнитель, ожидающий сериализованные tf.Example протоколы. |
receiver_tensors_alternatives | словарь из строки в дополнительные группы приемных тензоров, каждый из которых может быть Tensor, SparseTensor, или словарь из строки в Tensor или SparseTensor. Эти именованные альтернативы приемных тензоров генерируют дополнительные сигнатуры предоставления услуг, которые могут использоваться для подачи входных данных в различные моменты внутри подграфа приемника входных данных. Типичное использование — разрешить подачу исходных признаков Tensor ниже по потоку оператора tf.parse_example(). По умолчанию равно None. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/estimator/export/TensorServingInputReceiver
