src/transformers/models/data2vec/modeling_tf_data2vec_vision.py/TFData2VecVisionRelativePositionBias/call
class TFData2VecVisionRelativePositionBias:    def call(self, inputs=None) -> tf.Tensor:
        relative_position_bias = tf.gather(self.relative_position_bias_table, self.relative_position_index, axis=0)
        return tf.transpose(relative_position_bias, [2, 0, 1])