Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.graph_gradual_typechecker.first_two_eq#

torch.fx.experimental.graph_gradual_typechecker.first_two_eq(n)[source]#

For operations where the first two dimensions of the input and output shape are equal

On this page
Show Source
PyTorch Libraries