primeqa.ir.dense.colbert_top.colbert.search.strided_tensor.StridedTensor

class primeqa.ir.dense.colbert_top.colbert.search.strided_tensor.StridedTensor(packed_tensor, lengths, dim=None, use_gpu=False)

Bases: primeqa.ir.dense.colbert_top.colbert.search.strided_tensor_core.StridedTensorCore

Methods

as_packed_tensor
as_padded_tensor
as_tensors_list
from_nested_list
from_packed_tensor
from_padded_tensor
from_tensors_list
lookup
lookup_packed_unordered
lookup_staggered
pad_packed
try_load_torch_extensions