Functions — github.com/budzianowski/PyTorch-Beam-Search-Decoding

CodeHub MCPcopy Create free account

hub / github.com/budzianowski/PyTorch-Beam-Search-Decoding / functions

Functions6 in github.com/budzianowski/PyTorch-Beam-Search-Decoding

⨍ Functions 6 ◇ Types & classes 2

↓ 2 callersMethodeval

(self, alpha=1.0)

decode_beam.py:57

Illustrative decoder

decode_beam.py:14

:param hiddenstate: :param previousNode: :param wordId: :param logProb: :param length:

decode_beam.py:43

Functionbeam_decode

:param target_tensor: target indexes tensor of shape [B, T] where B is the batch size and T is the maximum length of the output sentence :par

decode_beam.py:67

(self, input, hidden, not_used)

decode_beam.py:29

Functiongreedy_decode

:param target_tensor: target indexes tensor of shape [B, T] where B is the batch size and T is the maximum length of the output sentence :par

decode_beam.py:163