TextTokenEncoder

class fairseq2.data.text.TextTokenEncoder[source]

Bases: ABC

Encodes text into tokens or token indices.

abstract __call__(text)[source]
Parameters:

text (str) – The text to encode.

Return type:

Tensor

abstract encode_as_tokens(text)[source]
Parameters:

text (str) – The text to encode.

Return type:

List[str]

abstract property prefix_indices: Tensor | None

Get the indices of the prefix tokens. Shape: \((S)\), where \(S\) is the number of indices.

abstract property suffix_indices: Tensor | None

Get the indices of the suffix tokens. Shape: \((S)\), where \(S\) is the number of indices.