How can I add a node at the end of the graph?

#1

Hello~ I have a graph that is from pytorch and I want to do some quantize with it, I want to add a node at the end of the graph, but i can’t find a proper way to do it. Could anyone tell me how to do it? Thanks~

#2

Suppose you obtain an IRModule mod from pytorch, and wangto add an nn.relu op at the end. You can do something like this:

func = mod["main"]
func_body = func.body
new_func_body = relay.nn.relu(func_body)
new_func = relay.Function(func.params, new_func_body)
new_mod = tvm.IRModule.from_expr(new_func)