Spec-Zone.ru › TensorFlow 2.4

tf.compat.v1.lite.OpHint.OpHintArgumentTracker

Понятийно отслеживает индексы аргументов «функций OpHint».

tf.compat.v1.lite.OpHint.OpHintArgumentTracker(
    function_name, unique_function_id, node_name_prefix, attr_name, level=1,
    children_inputs_mappings=None
)

Входы и аргументы этих функций оба используют экземпляр класса, чтобы они могли иметь независимую нумерацию.

Аргументы
function_name Имя функции, для которой отслеживаются аргументы.
unique_function_id UUID функции, для которой отслеживаются аргументы.
node_name_prefix Как называются создаваемые идентификаторы.
attr_name Имя атрибута, используемого для хранения индекса для этой подсказки. Например, FUNCTION_INPUT_INDEX или FUNCTION_OUTPUT_INDEX
level Иерархический уровень узла Ophint, число.
children_inputs_mappings Сопоставление входов/выходов для подсказок дочерних элементов.

Методы

add

Просмотреть исходный код

add(
    arg, tag=None, name=None, aggregate=None, index_override=None
)

Возвращает обернутый тензор тензора ввода в качестве аргумента.

Аргументы
arg Тензор TensorFlow, который должен рассматриваться как аргумент.
tag Строковый тег для идентификации аргументов, которые должны быть упакованы.
name Имя аргумента. Это включено в имена оператора подсказки Identity.
aggregate Стратегия агрегирования. Допустимые значения — OpHint.AGGREGATE_FIRST, OpHint.AGGREGATE_LAST и OpHint.AGGREGATE_STACK. Обратите внимание, что агрегирование допустимо только при указании тега.
index_override Укажите, какой индекс ввода/вывода должен быть в конечной заглушке. Например, add(arg0, index=1); add(arg1, index=0) приведет к тому, что конечная заглушка будет иметь вид stub_func(inputs[arg1, arg0], outputs=[]), а не в порядке вызова по умолчанию.
Возвращаемые значения
Тензор, представляющий обернутый аргумент.
Исключения
ValueError При несоответствии индексов.

© 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/compat/v1/lite/OpHint/OpHintArgumentTracker

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API