towhee.models.utils.get_relative_position_index.get_relative_position_index

towhee.models.utils.get_relative_position_index.get_relative_position_index(win_h: int, win_w: int) → Tensor[source]
Function to generate pair-wise relative position index for each token inside the window.

Taken from Timms Swin V1 implementation.

Parameters:

  • win_h (int) – Window/Grid height.

  • win_w (int) – Window/Grid width.

Returns:

Pair-wise relative position indexes [height * width, height * width].

Return type:

relative_coords (torch.Tensor)