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