def calculate_stability_score(masks: paddle.Tensor,
                              mask_threshold: float,
                              threshold_offset: float) -> paddle.Tensor:
    """
    Computes the stability score for a batch of masks. The stability
    score is the IoU between the binary masks obtained by thresholding
    the predicted mask logits at high and low values.
    """
    # One mask is always contained inside the other.
    # Save memory by preventing unnecessary cast to paddle.int64
    intersections = ((masks > (mask_threshold + threshold_offset)).cast('int16')
                     .sum(-1).cast('int32').sum(-1))
    unions = ((masks > (mask_threshold - threshold_offset)).cast('int16')
              .sum(-1).cast('int32').sum(-1))
    return intersections / unions

    Computes the stability score for a batch of masks. The stability
    score is the IoU between the binary masks obtained by thresholding
    the predicted mask logits at high and low values.
    
contrib/SegmentAnything/segment_anything/utils/amg.py