PackedTensor

class lightning_ir.retrieve.base.packed_tensor.PackedTensor(*args, lengths: Sequence[int] | None = None, **kwargs)

Bases: Tensor

__init__(*args, lengths: Sequence[int] | None = None, **kwargs) → None

Methods

__init__(*args[, lengths])
lookup(packed_idcs, idcs_lengths[, unique])
to_padded_tensor(padding[, output_size])	See to_padded_tensor()

Attributes

segmented_tensor

to_padded_tensor(padding, output_size=None) → Tensor

See to_padded_tensor()