megengine.traced_module.traced_module.InternalGraph.add_input_node

InternalGraph.add_input_node(shape, dtype='float32', name='args')

Add an input node to the graph.

The new Node will be the last of the positional arguments.

参数

• shape (Tuple[int]) – the shape of the new input Node.

• dtype (str) – the dtype of the new input Node. Default: float32

• name (str) – the name of the new input Node. When the name is used in the graph, a suffix will be added to it.

上一页

megengine.traced_module.traced_module.InternalGraph.top_graph

下一页

megengine.traced_module.traced_module.InternalGraph.add_output_node