我想使用 Mul 运算创建节点并将其添加到现有图表中。
这是我的尝试:
node1 = tf.NodeDef()
node1.name = 'MyMul1'
node1.op = 'Mul'
node1.input.extend(['conv1'])
node1.attr["T"].type = 1
我不知道如何在相乘的输入张量上指定常量的问题。
最佳答案
这可能不适用于您的情况，但我强烈建议使用更高级别的 API，例如 tf.math.multiply 。如果您丢失了相关输入的句柄，您可以使用 tf.Graph.get_tensor_by_name和 tf.Graph.get_operation_by_name .
也就是说，与常量相乘的方法是创建一个具有指定值的 Const 操作，然后使用它的输出作为该 Mul 的输入作品。 Example使用 tf.Graph.create_op 从 r1.13 创建 op 。不过，更高级别的 API 如 tf.constant可能更适合您的需求。
关于python - Tensorflow:如何使用 Mul 操作创建 tf.NodeDef()？，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/56411351/