megengine.traced_module.expr.Input

class Input(name=None, type=None, orig_name=None)

A fake Expr which is used to mark the input of graph.

Attributes

arg_def	The TreeDef used to reconstruct the input of the operation.
args	Get the the positional arguments of the operation corresponding to this Expr.
const_val	The non-tensor object in the input of the operation.
inputs	The input Nodes of this Expr.
kwargs	Get the the keyword arguments of the operation corresponding to this Expr.
name
out_def	The TreeDef used to reconstruct the output of the operation.
outputs	The output Nodes of this Expr.
top_graph	Get the parent graph of this Expr.

Methods

add_inputs(vals)
add_outputs(outputs)
disable_remove()
enable_remove()
make(*args, **kwargs)
replace_inputs(repl_dict)	Replace the input Nodes of this Expr.
unflatten_args(inputs)

上一页

megengine.traced_module.node.TensorNode.value

下一页

megengine.traced_module.expr.Input.arg_def