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

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

Bases: object

Methods

as_packed_tensor
as_padded_tensor
as_tensors_list
from_nested_list
from_packed_tensor
from_padded_tensor
from_tensors_list