Tensorflow: как создать tf.NodeDef () с операцией Mul?

Question

Я хочу создать узел с операцией Mul и добавить его в существующий график.

Вот моя попытка:

node1 = tf.NodeDef()
node1.name = 'MyMul1'
node1.op = 'Mul'
node1.input.extend(['conv1'])
node1.attr["T"].type = 1

Проблема в том, что я не знаю, как указать константу, на которую умножается входной тензор.

Answer 1

Это может быть неприменимо к вашей ситуации, но я настоятельно рекомендую использовать API более высокого уровня, такие как tf.math.multiply. Если вы потеряли дескриптор для рассматриваемых входов, вы можете использовать tf.Graph.get_tensor_by_name и tf.Graph.get_operation_by_name.

Тем не менее, что вы можете сделать для умножения на константу, это создать Const операцию с указанным значением и затем использовать ее вывод в качестве входных данных для этой Mul операции. Пример создания операции из r1.13 с использованием tf.Graph.create_op. Хотя API более высокого уровня, такой как tf.constant, может лучше удовлетворить ваши потребности.