EchoStateNetwork.forward#

EchoStateNetwork.forward(x, initial_state=None, return_states=False)[source]#

Forward pass of the network.

Parameters:

  • x (torch.Tensor) – input tensor.

  • initial_state (List[torch.Tensor]) – initial state of the reservoirs.

  • return_states (bool, optional) – whether to return the states. Defaults to False.

Returns:

output tensor.

Return type:

torch.Tensor