oneflow.nn.Graph.__call__

Graph.__call__(*args, **kwargs)

Call nn.Graph subclass instance to run your customized graph.

Call your customized graph after the instantiation:

For example:

g = CustomGraph()
out_tensors = g(input_tensors)

The inputs of __call__ method must match the inputs of build() method. And the __call__ method will return outputs matching the outputs of build() method.

Note

The first call takes longer than later calls, because nn.Graph will do the computaion graph generation and optimization at the first call.

Donot override this function.