def project_inner(self, index):
        """Projects an index with the same index set onto the inner components."""
        return IndexMap(
            indices=tf.math.floormod(index.indices, self.inner_index.num_segments),
            num_segments=self.inner_index.num_segments,
            batch_dims=index.batch_dims,
        )
Projects an index with the same index set onto the inner components.
dpkd/transformers/src/transformers/models/tapas/modeling_tf_tapas.py