 class="highlighttable">
         <tr>
          <td class="linenos">
           <div class="linenodiv">
            <pre> 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38</pre>
           </div>
          </td>
          <td class="code">
           <div class="highlight">
            <pre><span></span><span class="kn">import</span> <span class="nn">torch.nn</span> <span class="k">as</span> <span class="nn">nn</span>
<span class="kn">import</s