Я пытаюсь исследовать использование тензорного потока с пользовательскими операциями. Я создаю простой переключатель op и проверял его, как это предлагается в документе tensorflow. Теперь я пытаюсь построить график, а затем вызвать метод run() в тензорном потоке Сессия. Ниже мой код. Я получаю следующую ошибку. Может кто-то помочь, что я должен сделать, чтобы исправить это. Нужно ли переустанавливать tenorflow каждый раз, когда я добавляю новый пользовательский параметр op в /user_ops/?Построение графа с пользовательскими операциями в тензорном потоке
import tensorflow as tf
# Create a Constant op that produce integer value
input1 = tf.constant(10)
# Create another op that produce an integer value
input2 = tf.constant(5)
# Create op that produce 0 or 1 as the control input in a switch
input3 = tf.constant(1)
# Create a switch op that takes input1 and input2 as inputs and input3 as
# the control input to produce an output
out = tf.user_ops.simple_switch(input1, input2, input3)
# Launch a default graph
sess = tf.Session()
# Call the 'run()' method and get the result
result = sess.run(out)
print(result)
# Close the Session when we're done!
sess.close()
При выполнении интерпретатор питона я получаю следующее сообщение об ошибке:
Traceback (most recent call last): File "tensorflow-switch.py", line 14, in out = tf.simple_switch(input1, input2, input3) AttributeError: 'module' object has no attribute 'simple_switch'
Как выглядит отредактированный 'user_ops'? – Caridorc
@Caridorc Вы имеете в виду мой пользовательский op? – kishansudu
Я имею в виду: вы только что добавили свободную плавающую функцию к этому модулю или добавили ее в класс или словарь? – Caridorc