You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
<emclass="property"><spanclass="pre">class</span><spanclass="w"></span></em><spanclass="sig-prename descclassname"><spanclass="pre">torchjd.autogram.</span></span><spanclass="sig-name descname"><spanclass="pre">Engine</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">modules</span></span></em><spanclass="sig-paren">)</span><aclass="reference external" href="https://github.com/TorchJD/torchjd/blob/main/src/torchjd/autogram/_engine.py#L45-L198"><spanclass="viewcode-link"><spanclass="pre">[source]</span></span></a><aclass="headerlink" href="#torchjd.autogram.Engine" title="Link to this definition">¶</a></dt>
252
+
<emclass="property"><spanclass="pre">class</span><spanclass="w"></span></em><spanclass="sig-prename descclassname"><spanclass="pre">torchjd.autogram.</span></span><spanclass="sig-name descname"><spanclass="pre">Engine</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">modules</span></span></em><spanclass="sig-paren">)</span><aclass="reference external" href="https://github.com/TorchJD/torchjd/blob/main/src/torchjd/autogram/_engine.py#L45-L202"><spanclass="viewcode-link"><spanclass="pre">[source]</span></span></a><aclass="headerlink" href="#torchjd.autogram.Engine" title="Link to this definition">¶</a></dt>
253
253
<dd><p>Used for computing the Gramian of the Jacobian of some vector with respect to the direct
254
254
parameters of all provided modules.</p>
255
255
<p>After this object is constructed, the outputs of the provided modules will have an extended
@@ -330,7 +330,7 @@ <h1>Engine<a class="headerlink" href="#engine" title="Link to this heading">¶</
<spanclass="sig-name descname"><spanclass="pre">compute_gramian</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">output</span></span></em><spanclass="sig-paren">)</span><aclass="reference external" href="https://github.com/TorchJD/torchjd/blob/main/src/torchjd/autogram/_engine.py#L163-L172"><spanclass="viewcode-link"><spanclass="pre">[source]</span></span></a><aclass="headerlink" href="#torchjd.autogram.Engine.compute_gramian" title="Link to this definition">¶</a></dt>
333
+
<spanclass="sig-name descname"><spanclass="pre">compute_gramian</span></span><spanclass="sig-paren">(</span><emclass="sig-param"><spanclass="n"><spanclass="pre">output</span></span></em><spanclass="sig-paren">)</span><aclass="reference external" href="https://github.com/TorchJD/torchjd/blob/main/src/torchjd/autogram/_engine.py#L163-L183"><spanclass="viewcode-link"><spanclass="pre">[source]</span></span></a><aclass="headerlink" href="#torchjd.autogram.Engine.compute_gramian" title="Link to this definition">¶</a></dt>
334
334
<dd><p>Compute the Gramian of the Jacobian of <codeclass="docutils literal notranslate"><spanclass="pre">output</span></code> with respect the direct parameters of all
0 commit comments