Rate this Page
★ ★ ★ ★ ★

torch.fx.experimental.migrate_gradual_types.constraint_transformation.transform_index_select#

torch.fx.experimental.migrate_gradual_types.constraint_transformation.transform_index_select(constraint, counter)[source]#

The constraints consider the given tensor size, checks if the index is valid and if so, generates a constraint for replacing the input dimension with the required dimension

On this page
Show Source
PyTorch Libraries