Function divide

SwissArmyTransformer/sat/mpu/utils.py:27–31

Ensure that numerator is divisible by the denominator and return the division value.

(numerator, denominator)

def divide(numerator, denominator):
    """Ensure that numerator is divisible by the denominator and return
    the division value."""
    ensure_divisibility(numerator, denominator)
    return numerator // denominator


def split_tensor_along_last_dim(tensor, num_partitions,