PINTO

<ul>
<li>512x736</li>
</ul>
<div class="code-block-container">
<div class="code-block-filename-container">mlp_classification</div>
<pre class="language-python"><code class="language-python">features.shape
torch.Size([1, 96, 50000])
</code></pre>
</div><div class="code-block-container">
<div class="code-block-filename-container">self.probs</div>
<pre class="language-python"><code class="language-python">self.probs.shape
torch.Size([1, 256, 50000])
</code></pre>
</div><div class="code-block-container">
<div class="code-block-filename-container">mlp_regression</div>
<pre class="language-python"><code class="language-python">torch.cat([features, self.disparity], 1).shape
torch.Size([1, 97, 50000])
</code></pre>
</div><div class="code-block-container">
<div class="code-block-filename-container">offset</div>
<pre class="language-python"><code class="language-python">offset.shape
torch.Size([1, 1, 50000])
</code></pre>
</div><div class="code-block-container">
<div class="code-block-filename-container">self.disparity</div>
<pre class="language-python"><code class="language-python">(self.disparity + offset).shape
torch.Size([1, 1, 50000])
</code></pre>
</div>