some questions about the attention

[haoren55555]

Could anyone tells me what this sentence means :
attn_qs = attn[torch.arange(B)[:, None], patch_id, :];
why use "torch.arange(B)"