Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.graph_gradual_typechecker.transpose_inference_rule#

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

We check that dimensions for the transpose operations are within range of the tensor type of the node

On this page
Show Source
PyTorch Libraries