Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.graph_gradual_typechecker.flatten_check#

torch.fx.experimental.graph_gradual_typechecker.flatten_check(tensor_type, start_dim, end_dim)[source]#
Return type:

TensorType

On this page
Show Source
PyTorch Libraries