BiEncoderEmbedding

class lightning_ir.bi_encoder.model.BiEncoderEmbedding(embeddings: torch.Tensor, scoring_mask: torch.Tensor)

Bases: object

__init__(embeddings: Tensor, scoring_mask: Tensor) → None

Methods

__init__(embeddings, scoring_mask)
items()
to()

Attributes

device
embeddings
scoring_mask