espnet.nets.scorer_interface.BatchPartialScorerInterface

source

class espnet.nets.scorer_interface.BatchPartialScorerInterface

Bases: BatchScorerInterface, PartialScorerInterface

Batch partial scorer interface for beam search.

batch_score_partial(ys: Tensor, next_tokens: Tensor, states: List[Any], xs: Tensor) → Tuple[Tensor, Any]

Score new token (required).

• Parameters:
  • ys (torch.Tensor) – torch.int64 prefix tokens (n_batch, ylen).
  • next_tokens (torch.Tensor) – torch.int64 tokens to score (n_batch, n_token).
  • states (List *[*Any ]) – Scorer states for prefix tokens.
  • xs (torch.Tensor) – The encoder feature that generates ys (n_batch, xlen, n_feat).
• Returns: Tuple of a score tensor for ys that has a shape (n_batch, n_vocab) and next states for ys
• Return type: tuple[torch.Tensor, Any]