Получение списка зависимостей от оптимизатора назад в тензорном потоке [дубликат]

Как насчет этого?

parseFloat(input.substring(5,10));

Где 5,10 - начальные и конечные смещения нужного фрагмента строки.

2
задан Kh40tiK 7 August 2017 в 17:44
поделиться

1 ответ

Нет встроенной функции (о которой я знаю), но ее легко сделать:

# Setup a graph
import tensorflow as tf
placeholder0 = tf.placeholder(tf.float32, [])
placeholder1 = tf.placeholder(tf.float32, [])
constant0 = tf.constant(2.0)
sum0 = tf.add(placeholder0, constant0)
sum1 = tf.add(placeholder1, sum0)

# Function to get *all* dependencies of a tensor.
def get_dependencies(tensor):
    dependencies = set()
    dependencies.update(tensor.op.inputs)
    for sub_op in tensor.op.inputs:
        dependencies.update(get_dependencies(sub_op))
    return dependencies

print(get_dependencies(sum0))
print(get_dependencies(sum1))
# Filter on type to get placeholders.
print([tensor for tensor in get_dependencies(sum0) if tensor.op.type == 'Placeholder'])
print([tensor for tensor in get_dependencies(sum1) if tensor.op.type == 'Placeholder'])

Конечно, вы также можете включить фильтрацию заполнителя в функцию.

3
ответ дан golmschenk 17 August 2018 в 10:41
поделиться
  • 1
    Эта рекурсия, вероятно, неэффективна в отношении «Фибоначчи». как графы, но в действительности хорошая начальная точка. – Kh40tiK 9 August 2017 в 01:31
  • 2
    если есть только одна зависимость от заполнителя, скажем y1 = x, где x является заполнителем, то эта функция get_depednececy не может ее найти. – Shaowu 4 August 2018 в 16:14
Другие вопросы по тегам:

Похожие вопросы: