Tensorflow：如何使用Mul操作创建tf.NodeDef（）？

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 op，然后将其输出用作此Mul op的输入。 Example使用tf.Graph.create_op从r1.13创建op。虽然，像tf.constant这样的高级API可能会更好地满足您的需求。