python Tensorflow:创建tf.NodeDef()并设置属性

mwngjboj  于 2023-03-21  发布在  Python
关注(0)|答案(3)|浏览(105)

我正在尝试创建一个新节点并设置其属性。
例如,打印一个图形节点,我看到它的属性是:

attr {
  key: "T"
  value {
    type: DT_FLOAT
  }
}

我可以创建一个节点,如下所示:

node = tf.NodeDef(name='MyConstTensor', op='Const',
                   attr={'value': tf.AttrValue(tensor=tensor_proto),
                         'dtype': tf.AttrValue(type=dt)})

但是如何添加key: "T"属性?即在这种情况下,tf.AttrValue内部应该是什么?
attr_value.proto我试过了:

node = tf.NodeDef()
node.name = 'MySub'
node.op = 'Sub'
node.input.extend(['MyConstTensor', 'conv2'])
node.attr["key"].s = 'T' # TypeError: 'T' has type str, but expected one of: bytes

更新日期:

我发现在Tensorflow中应该这样写:

node.attr["T"].type = b'float32'

但这给出了一个错误:
TypeError:b'float32'具有类型字节,但应为以下之一:整型、长整型
我不确定哪个int值对应于float32。
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/framework/attr_value.proto#L23
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/framework/attr_value.proto#L35

tmb3ates

tmb3ates1#

通过反复试验,我得出结论:

node.attr["T"].type = 1 # to set type to float32
bihw5rsg

bihw5rsg2#

尝试将T作为字节传递:

node.attr["key"].s = b'T'

如果你想传递更多的字符,试试bytearray类。
在protobuf中,AttrValue的s被定义为bytes而不是string。Protobuf手册指出这应该是python中的字符串,但您的错误表明它更像是一个字节数组。

r3i60tvu

r3i60tvu3#

不需要尝试和错误:

node.attr["T"].type = tf.float32.as_datatype_enum

总的来说,您可以在tensorflow/c/tf_datatype.h(参见typedef enum TF_DataType)中看到源代码中的所有数据类型代码。

相关问题