8 lines
1002 B
Plaintext
8 lines
1002 B
Plaintext
\begin{MintedVerbatim}[commandchars=\\\{\}]
|
|
\PYG{k}{fn}\PYG{+w}{ }\PYG{n+nf}{forward}\PYG{p}{(}\PYG{o}{\PYGZam{}}\PYG{n+nb+bp}{self}\PYG{p}{,}\PYG{+w}{ }\PYG{n}{x}\PYG{p}{:}\PYG{+w}{ }\PYG{n+nc}{Tensor}\PYG{o}{\PYGZlt{}}\PYG{n}{B}\PYG{p}{,}\PYG{+w}{ }\PYG{l+m+mi}{2}\PYG{o}{\PYGZgt{}}\PYG{p}{)}\PYG{+w}{ }\PYG{p}{\PYGZhy{}\PYGZgt{}}\PYG{+w}{ }\PYG{n+nc}{Tensor}\PYG{o}{\PYGZlt{}}\PYG{n}{B}\PYG{p}{,}\PYG{+w}{ }\PYG{l+m+mi}{2}\PYG{o}{\PYGZgt{}}\PYG{+w}{ }\PYG{p}{\PYGZob{}}
|
|
\PYG{+w}{ }\PYG{k+kd}{let}\PYG{+w}{ }\PYG{n}{x}\PYG{+w}{ }\PYG{o}{=}\PYG{+w}{ }\PYG{n+nb+bp}{self}\PYG{p}{.}\PYG{n}{linear1}\PYG{p}{.}\PYG{n}{forward}\PYG{p}{(}\PYG{n}{x}\PYG{p}{);}
|
|
\PYG{+w}{ }\PYG{k+kd}{let}\PYG{+w}{ }\PYG{n}{x}\PYG{+w}{ }\PYG{o}{=}\PYG{+w}{ }\PYG{n+nb+bp}{self}\PYG{p}{.}\PYG{n}{activation}\PYG{p}{.}\PYG{n}{forward}\PYG{p}{(}\PYG{n}{x}\PYG{p}{);}
|
|
\PYG{+w}{ }\PYG{n+nb+bp}{self}\PYG{p}{.}\PYG{n}{linear2}\PYG{p}{.}\PYG{n}{forward}\PYG{p}{(}\PYG{n}{x}\PYG{p}{)}
|
|
\PYG{p}{\PYGZcb{}}
|
|
\end{MintedVerbatim}
|