Rate this Page

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