cat_arange

lightning_ir.retrieve.base.searcher.cat_arange(arange_starts: Tensor, arange_ends: Tensor) → Tensor[source]

Concatenates arange tensors into a single tensor.

Parameters:

  • arange_starts (torch.Tensor) – The start indices of the ranges.

  • arange_ends (torch.Tensor) – The end indices of the ranges.

Returns:

A tensor containing the concatenated ranges.

Return type:

torch.Tensor