Rate this Page

torch.fx.experimental.migrate_gradual_types.constraint_transformation.is_target_div_by_dim#

torch.fx.experimental.migrate_gradual_types.constraint_transformation.is_target_div_by_dim(target, dim)[source]#

Generate constraints to check if the target dimensions are divisible by the input dimensions :param target: Target dimensions :param dim: Input dimensions

Returns: Constraints to check divisibility