Class IrVarNode

python/paddle/base/framework.py:5322–5418

Python IrVarNode. Beneath it is a core.Node, it inherits from IrNode.

class IrVarNode(IrNode):
    """
    Python IrVarNode. Beneath it is a core.Node, it inherits from IrNode.
    """

    def __init__(self, node):
        """
        Construct an IrVarNode using core.Node.

        Args:
            node(core.Node): C++ Node.
        """
        assert isinstance(node, core.Node) and node.is_var(), (
            "node must be the instance of core.Node and it must be a variable node."
        )
        super().__init__(node)
        self.node = node

    def set_shape(self, shape):
        """
        Set the node variable shape.

        Args:
            shape(list): shape to be set.
        """
        assert self.node.var() is not None, (
            "The node variable description can not be None."
        )
        self.node.var().set_shape(shape)

    def persistable(self):
        """
        If the variable node is a persistable variable, then return true.

        Returns:
            bool: indicate whether the variable is persistable.
        """
        assert self.node.var() is not None, (
            "The node variable description can not be None."
        )
        return self.node.var().persistable()

    def type(self):
        """
        Return the variable type.

        Returns:
            core.VarDesc.VarType: the variable type.
        """
        assert self.node.var() is not None, (
            "The node variable description can not be None."
        )
        return self.node.var().type()

    def dtype(self):
        """
        Return the variable data type.