torch.fx.experimental.migrate_gradual_types.constraint_transformation.is_dim_div_by_target#
- torch.fx.experimental.migrate_gradual_types.constraint_transformation.is_dim_div_by_target(target, dim)[source]#
Generate constraints to check if the input dimensions is divisible by the target dimensions :param target: Target dimensions :param dim: Input dimensions
Returns: Constraints to check divisibility