\frametitle{Forward pass}
\begin{lstlisting}[language=Rust, style=colouredRust]
fn forward(&self, x: Tensor<B, 2>) -> Tensor<B, 2> {
    let x = self.linear1.forward(x);
    let x = self.activation.forward(x);
    self.linear2.forward(x)
}
\end{lstlisting}