10 lines
258 B
Plaintext
10 lines
258 B
Plaintext
\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}
|
|
|