Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.graph_gradual_typechecker.relu_inference_rule#

torch.fx.experimental.graph_gradual_typechecker.relu_inference_rule(n, module_instance)[source]#

Input and output shapes should be equal.

Return type:

Any

On this page
Show Source
PyTorch Libraries