PackedTensor

class lightning_ir.retrieve.plaid.packed_tensor.PackedTensor(packed_tensor: Tensor, lengths: Sequence[int])

Bases: object

__init__(packed_tensor: Tensor, lengths: Sequence[int]) → None

Methods

__init__(packed_tensor, lengths)
lookup(packed_idcs, idcs_lengths[, unique])
to_padded_tensor([pad_value])

Attributes

segmented_tensor