使用tensorflow.python.framework.graph_util模块中的tensor_shape_from_node_def_name()函数解析张量维度

tensorflow.python.framework.graph_util模块中的tensor_shape_from_node_def_name()函数用于解析张量（Tensor）的维度信息。该函数的定义如下：

def tensor_shape_from_node_def_name(graph_def, name):
    """
    Returns a list representing the shape of a tensor.

    Args:
        graph_def: The GraphDef protocol buffer that contains the tensor.
        name: The name of the tensor.

    Returns:
        A list representing the shape of the tensor. Each element of the list
        represents the size of the corresponding dimension of the tensor.
    """
    node_def = find_node_def(name, graph_def)
    if node_def is None:
        raise ValueError("Node with name '{}' not found.".format(name))
    
    if node_def.op == 'Const':
        for attr in node_def.attr.values():
            if attr.tensor.tensor_content:
                shape_proto = attr.tensor.tensor_shape
                if shape_proto.dim:
                    return [dim.size for dim in shape_proto.dim]
    elif node_def.op == 'Placeholder':
        for attr in node_def.attr.values():
            if attr.type == 'DT_INT32':
                shape_proto = attr.shape
                if shape_proto.dim:
                    return [dim.size for dim in shape_proto.dim]
    
    raise ValueError("Invalid node type for tensor '{}'.".format(name))

该函数接收两个参数：graph_def和name。graph_def是包含张量信息的GraphDef协议缓冲区，name是要解析的张量的名称。函数将返回一个列表，表示张量的形状。列表中的每个元素表示张量相应维度的大小。

下面是一个使用tensor_shape_from_node_def_name()函数解析张量维度的示例：

import tensorflow as tf
from tensorflow.python.framework import graph_util

# 构建一个简单的计算图
a = tf.constant([1, 2, 3], name='a')
b = tf.constant([4, 5, 6], name='b')
c = tf.add(a, b, name='c')

# 获取默认图
graph = tf.get_default_graph()

# 导出计算图的GraphDef
graph_def = graph.as_graph_def()

# 获取张量c的维度
shape = graph_util.tensor_shape_from_node_def_name(graph_def, 'c')
print('Shape of tensor c:', shape)

在上面的示例中，首先构建了一个简单的计算图，在该计算图中，定义了两个常量张量a和b，以及一个将a和b相加的操作c。然后通过tf.get_default_graph()获取默认计算图并导出计算图的GraphDef。最后，使用tensor_shape_from_node_def_name()函数获取了张量c的维度，结果打印出来。

注意，这里的graph_util.tensor_shape_from_node_def_name()函数是通过tensorflow.python.framework.graph_util模块导入的。