Decoder.forward

Decoder.forward(inputs: Dict[str, torch.Tensor]) Dict[str, torch.Tensor][source]

Forward pass of the decoder.

Parameters:

inputs (Dict[str, torch.Tensor]) – Dictionary of input tensors, where the keys are strings and the values are PyTorch tensors.

Returns:

Dict[str, torch.Tensor] – Dictionary of output tensors, where the keys are strings and the values are PyTorch tensors.