computational_graph.py 文件源码

def __init__(self, node, attribute=None):
        """Initializes DotNode.

        Args:
            node: :class: `Variable` object or :class: `Function` object.
            attribute (dict): Attributes for the node.

        """
        assert isinstance(node, (variable.Variable, function.Function))
        self.node = node
        self.id_ = id(node)
        self.attribute = {'label': node.label}
        if isinstance(node, variable.Variable):
            self.attribute = {'shape': 'oval'}
        else:
            self.attribute = {'shape': 'box'}
        if attribute is not None:
            self.attribute.update(attribute)