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. |
© 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.4/api_docs/python/tf/estimator/export/TensorServingInputReceiver
