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.