Deploying to gh-pages from @ 4b9697e 🚀

intsystems · Dec 6, 2024 · f32eb57 · f32eb57
1 parent 1c92323
commit f32eb57
Show file tree

Hide file tree

Showing 16 changed files with 55 additions and 829 deletions.
diff --git a/.doctrees/environment.pickle b/.doctrees/environment.pickle
diff --git a/.doctrees/relaxit.distributions.doctree b/.doctrees/relaxit.distributions.doctree
diff --git a/_modules/relaxit/distributions/CorrelatedRelaxedBernoulli.html b/_modules/relaxit/distributions/CorrelatedRelaxedBernoulli.html
@@ -95,7 +95,7 @@ <h1>Source code for relaxit.distributions.CorrelatedRelaxedBernoulli</h1><div cl
 
 
 <div class="viewcode-block" id="CorrelatedRelaxedBernoulli">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli">[docs]</a>
 <span class="k">class</span> <span class="nc">CorrelatedRelaxedBernoulli</span><span class="p">(</span><span class="n">TorchDistribution</span><span class="p">):</span>
 <span class="w">    </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">    Correlated Relaxed Bernoulli distribution class inheriting from Pyro&#39;s TorchDistribution.</span>
@@ -168,7 +168,7 @@ <h1>Source code for relaxit.distributions.CorrelatedRelaxedBernoulli</h1><div cl
         <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">pi</span><span class="o">.</span><span class="n">shape</span><span class="p">[</span><span class="o">-</span><span class="mi">1</span><span class="p">:]</span>
 
 <div class="viewcode-block" id="CorrelatedRelaxedBernoulli.rsample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.rsample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli.rsample">[docs]</a>
     <span class="k">def</span> <span class="nf">rsample</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sample_shape</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">())</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution using the reparameterization trick.</span>
@@ -200,7 +200,7 @@ <h1>Source code for relaxit.distributions.CorrelatedRelaxedBernoulli</h1><div cl
 
 
 <div class="viewcode-block" id="CorrelatedRelaxedBernoulli.sample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.sample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli.sample">[docs]</a>
     <span class="k">def</span> <span class="nf">sample</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sample_shape</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">())</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution.</span>
@@ -215,7 +215,7 @@ <h1>Source code for relaxit.distributions.CorrelatedRelaxedBernoulli</h1><div cl
 
 
 <div class="viewcode-block" id="CorrelatedRelaxedBernoulli.log_prob">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.log_prob">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.CorrelatedRelaxedBernoulli.CorrelatedRelaxedBernoulli.log_prob">[docs]</a>
     <span class="k">def</span> <span class="nf">log_prob</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Computes the log probability of the given value.</span>

diff --git a/_modules/relaxit/distributions/GaussianRelaxedBernoulli.html b/_modules/relaxit/distributions/GaussianRelaxedBernoulli.html
@@ -95,7 +95,7 @@ <h1>Source code for relaxit.distributions.GaussianRelaxedBernoulli</h1><div clas
 
 
 <div class="viewcode-block" id="GaussianRelaxedBernoulli">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli">[docs]</a>
 <span class="k">class</span> <span class="nc">GaussianRelaxedBernoulli</span><span class="p">(</span><span class="n">TorchDistribution</span><span class="p">):</span>
 <span class="w">    </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">    Gaussian-based continuous Relaxed Bernoulli distribution class inheriting from Pyro&#39;s TorchDistribution.</span>
@@ -148,7 +148,7 @@ <h1>Source code for relaxit.distributions.GaussianRelaxedBernoulli</h1><div clas
         <span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">()</span>
 
 <div class="viewcode-block" id="GaussianRelaxedBernoulli.rsample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.rsample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli.rsample">[docs]</a>
     <span class="k">def</span> <span class="nf">rsample</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sample_shape</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">())</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution using the reparameterization trick.</span>
@@ -164,7 +164,7 @@ <h1>Source code for relaxit.distributions.GaussianRelaxedBernoulli</h1><div clas
 
 
 <div class="viewcode-block" id="GaussianRelaxedBernoulli.sample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.sample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli.sample">[docs]</a>
     <span class="k">def</span> <span class="nf">sample</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sample_shape</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">())</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution.</span>
@@ -179,7 +179,7 @@ <h1>Source code for relaxit.distributions.GaussianRelaxedBernoulli</h1><div clas
 
 
 <div class="viewcode-block" id="GaussianRelaxedBernoulli.log_prob">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.log_prob">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli.log_prob">[docs]</a>
     <span class="k">def</span> <span class="nf">log_prob</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Computes the log probability of the given value.</span>

diff --git a/_modules/relaxit/distributions/GumbelSoftmaxTopK.html b/_modules/relaxit/distributions/GumbelSoftmaxTopK.html
@@ -84,7 +84,7 @@ <h1>Source code for relaxit.distributions.GumbelSoftmaxTopK</h1><div class="high
 
 
 <div class="viewcode-block" id="GumbelSoftmaxTopK">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK">[docs]</a>
 <span class="k">class</span> <span class="nc">GumbelSoftmaxTopK</span><span class="p">(</span><span class="n">TorchDistribution</span><span class="p">):</span>
 <span class="w">    </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">    Implementation of the Gaussian-Softmax top-K trick from https://arxiv.org/pdf/1903.06059.</span>
@@ -164,7 +164,7 @@ <h1>Source code for relaxit.distributions.GumbelSoftmaxTopK</h1><div class="high
         <span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">()</span>
 
 <div class="viewcode-block" id="GumbelSoftmaxTopK.rsample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.rsample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK.rsample">[docs]</a>
     <span class="k">def</span> <span class="nf">rsample</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution using the Gaussian-soft max topK trick.</span>
@@ -184,7 +184,7 @@ <h1>Source code for relaxit.distributions.GumbelSoftmaxTopK</h1><div class="high
 
 
 <div class="viewcode-block" id="GumbelSoftmaxTopK.sample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.sample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK.sample">[docs]</a>
     <span class="k">def</span> <span class="nf">sample</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution with no grad.</span>
@@ -197,7 +197,7 @@ <h1>Source code for relaxit.distributions.GumbelSoftmaxTopK</h1><div class="high
 
 
 <div class="viewcode-block" id="GumbelSoftmaxTopK.log_prob">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.log_prob">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.GumbelSoftmaxTopK.GumbelSoftmaxTopK.log_prob">[docs]</a>
     <span class="k">def</span> <span class="nf">log_prob</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Computes the log probability of the given value.</span>

diff --git a/_modules/relaxit/distributions/HardConcrete.html b/_modules/relaxit/distributions/HardConcrete.html
@@ -82,7 +82,7 @@ <h1>Source code for relaxit.distributions.HardConcrete</h1><div class="highlight
 
 
 <div class="viewcode-block" id="HardConcrete">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.HardConcrete">[docs]</a>
 <span class="k">class</span> <span class="nc">HardConcrete</span><span class="p">(</span><span class="n">TorchDistribution</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
 
@@ -146,7 +146,7 @@ <h1>Source code for relaxit.distributions.HardConcrete</h1><div class="highlight
         <span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">()</span>
 
 <div class="viewcode-block" id="HardConcrete.rsample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.rsample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.HardConcrete.rsample">[docs]</a>
     <span class="k">def</span> <span class="nf">rsample</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sample_shape</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">())</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution using the reparameterization trick.</span>
@@ -166,7 +166,7 @@ <h1>Source code for relaxit.distributions.HardConcrete</h1><div class="highlight
 
 
 <div class="viewcode-block" id="HardConcrete.sample">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.sample">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.HardConcrete.sample">[docs]</a>
     <span class="k">def</span> <span class="nf">sample</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sample_shape</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</span><span class="n">Size</span><span class="p">())</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Generates a sample from the distribution.</span>
@@ -239,7 +239,7 @@ <h1>Source code for relaxit.distributions.HardConcrete</h1><div class="highlight
         <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_Q_prob</span><span class="p">((</span><span class="n">value</span> <span class="o">-</span> <span class="bp">self</span><span class="o">.</span><span class="n">gamma</span><span class="p">)</span> <span class="o">/</span> <span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">xi</span> <span class="o">-</span> <span class="bp">self</span><span class="o">.</span><span class="n">gamma</span><span class="p">))</span>
 
 <div class="viewcode-block" id="HardConcrete.log_prob">
-<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.log_prob">[docs]</a>
+<a class="viewcode-back" href="../../../relaxit.distributions.html#relaxit.distributions.HardConcrete.HardConcrete.log_prob">[docs]</a>
     <span class="k">def</span> <span class="nf">log_prob</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">value</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">torch</span><span class="o">.</span><span class="n">Tensor</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;</span>
 <span class="sd">        Computes the log probability of the given value.</span>