src/transformers/models/tapas/modeling_tf_tapas.py/ProductIndexMap/project_outer
class ProductIndexMap:    def project_outer(self, index):
        """Projects an index with the same index set onto the outer components."""
        return IndexMap(
            indices=tf.math.floordiv(index.indices, self.inner_index.num_segments),
            num_segments=self.outer_index.num_segments,
            batch_dims=index.batch_dims,
        )