Encoder.forward

Encoder.forward(x) Dict[str, torch.Tensor][source]

Forward pass of the encoder.

Parameters:

x (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.