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
<li><p><strong>pref_vector</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.11)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a> | <aclass="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.14)"><codeclass="xref py py-obj docutils literal notranslate"><spanclass="pre">None</span></code></a></span>) – The preference vector to use. If not provided, defaults to
312
+
<li><p><strong>pref_vector</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.12)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a> | <aclass="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.14)"><codeclass="xref py py-obj docutils literal notranslate"><spanclass="pre">None</span></code></a></span>) – The preference vector to use. If not provided, defaults to
312
313
<spanclass="math notranslate nohighlight">\(\begin{bmatrix} \frac{1}{m} & \dots & \frac{1}{m} \end{bmatrix}^T \in \mathbb{R}^m\)</span>.</p></li>
313
314
<li><p><strong>scale_mode</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.python.org/3/library/typing.html#typing.Literal" title="(in Python v3.14)"><codeclass="xref py py-data docutils literal notranslate"><spanclass="pre">Literal</span></code></a>[<codeclass="docutils literal notranslate"><spanclass="pre">'min'</span></code>, <codeclass="docutils literal notranslate"><spanclass="pre">'median'</span></code>, <codeclass="docutils literal notranslate"><spanclass="pre">'rmse'</span></code>]</span>) – The scaling mode used to build the balance transformation. <codeclass="docutils literal notranslate"><spanclass="pre">"min"</span></code> uses
314
315
the smallest eigenvalue (default), <codeclass="docutils literal notranslate"><spanclass="pre">"median"</span></code> uses the median eigenvalue, and <codeclass="docutils literal notranslate"><spanclass="pre">"rmse"</span></code>
@@ -327,10 +328,10 @@ <h1>Aligned-MTL<a class="headerlink" href="#aligned-mtl" title="Link to this hea
327
328
<dd><p>Computes the aggregation from the input matrix and applies all registered hooks.</p>
<li><p><strong>pref_vector</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.11)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a> | <aclass="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.14)"><codeclass="xref py py-obj docutils literal notranslate"><spanclass="pre">None</span></code></a></span>) – The preference vector to use. If not provided, defaults to
349
+
<li><p><strong>pref_vector</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.12)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a> | <aclass="reference external" href="https://docs.python.org/3/library/constants.html#None" title="(in Python v3.14)"><codeclass="xref py py-obj docutils literal notranslate"><spanclass="pre">None</span></code></a></span>) – The preference vector to use. If not provided, defaults to
349
350
<spanclass="math notranslate nohighlight">\(\begin{bmatrix} \frac{1}{m} & \dots & \frac{1}{m} \end{bmatrix}^T \in \mathbb{R}^m\)</span>.</p></li>
350
351
<li><p><strong>scale_mode</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.python.org/3/library/typing.html#typing.Literal" title="(in Python v3.14)"><codeclass="xref py py-data docutils literal notranslate"><spanclass="pre">Literal</span></code></a>[<codeclass="docutils literal notranslate"><spanclass="pre">'min'</span></code>, <codeclass="docutils literal notranslate"><spanclass="pre">'median'</span></code>, <codeclass="docutils literal notranslate"><spanclass="pre">'rmse'</span></code>]</span>) – The scaling mode used to build the balance transformation. <codeclass="docutils literal notranslate"><spanclass="pre">"min"</span></code> uses
351
352
the smallest eigenvalue (default), <codeclass="docutils literal notranslate"><spanclass="pre">"median"</span></code> uses the median eigenvalue, and <codeclass="docutils literal notranslate"><spanclass="pre">"rmse"</span></code>
@@ -359,10 +360,10 @@ <h1>Aligned-MTL<a class="headerlink" href="#aligned-mtl" title="Link to this hea
359
360
<dd><p>Computes the vector of weights from the input Gramian and applies all registered hooks.</p>
<ddclass="field-odd"><p><strong>gramian</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.11)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The Gramian from which the weights must be extracted.</p>
363
+
<ddclass="field-odd"><p><strong>gramian</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.12)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The Gramian from which the weights must be extracted.</p>
<ddclass="field-odd"><p><strong>weights</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.11)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The weights associated to the rows of the input matrices.</p>
311
+
<ddclass="field-odd"><p><strong>weights</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.12)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The weights associated to the rows of the input matrices.</p>
311
312
</dd>
312
313
</dl>
313
314
<dlclass="py method">
@@ -316,10 +317,10 @@ <h1>Constant<a class="headerlink" href="#constant" title="Link to this heading">
316
317
<dd><p>Computes the aggregation from the input matrix and applies all registered hooks.</p>
<ddclass="field-odd"><p><strong>weights</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.11)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The weights to return at each call.</p>
337
+
<ddclass="field-odd"><p><strong>weights</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.12)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The weights to return at each call.</p>
337
338
</dd>
338
339
</dl>
339
340
<dlclass="py method">
@@ -342,10 +343,10 @@ <h1>Constant<a class="headerlink" href="#constant" title="Link to this heading">
342
343
<dd><p>Computes the vector of weights from the input matrix and applies all registered hooks.</p>
<ddclass="field-odd"><p><strong>matrix</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.11)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The matrix from which the weights must be extracted.</p>
346
+
<ddclass="field-odd"><p><strong>matrix</strong> (<spanclass="sphinx_autodoc_typehints-type"><aclass="reference external" href="https://docs.pytorch.org/docs/stable/tensors.html#torch.Tensor" title="(in PyTorch v2.12)"><codeclass="xref py py-class docutils literal notranslate"><spanclass="pre">Tensor</span></code></a></span>) – The matrix from which the weights must be extracted.</p>
0 commit comments