Как получить tenorflow op по имени?
Вы можете получить тензор по имени tf.get_default_graph().get_tensor_by_name("tensor_name:0")
Но можете ли вы получить операцию, например Optimizer.minimize
или enqueue
в очереди?
В моей первой модели я вернул все тензоры и ops, которые мне понадобятся из функции build_model
. Но список тензоров стал уродливым. В более поздних моделях я бросил все тензоры и ops в словаре для облегчения доступа. На этот раз мне показалось, что я просто ищу тензоры по имени, поскольку они мне нужны, но я не знаю, как это сделать с операциями.
Или есть лучший способ сделать это? Я нахожу различные тензоры и ops необходимы повсюду. Тренинг, код вывода, тестовые примеры, следовательно, желание иметь хороший стандартный способ доступа к различным частям графика без передачи переменных по всему месту.
Ответы
Ответ 1
Вы можете использовать метод tf.Graph.get_operation_by_name()
, чтобы получить tf.Operation
по имени. Например, чтобы получить операцию с именем "enqueue"
из графика по умолчанию:
op = tf.get_default_graph().get_operation_by_name("enqueue")