mwptoolkit.module.Strategy.greedy

mwptoolkit.module.Strategy.greedy.greedy_search(logits)

Find the index of max logits

Parameters

logits (torch.Tensor) – logits distribution

Returns

the chosen index of token

Return type

torch.Tensor