Generate Python docs from pytorch/pytorch@2b1ba0c

Author: pytorchbot

docs/main/_images/RReLU.png

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/index.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/__config__.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/__future__.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_jit_internal.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_lobpcg.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_logging/_internal.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 
@@ -598,7 +598,8 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
 <span class="sd">                               integer, and defaults to 2</span>
 
 <span class="sd">        M (Tensor, optional): the input tensor&#39;s mean of size</span>
-<span class="sd">                              :math:`(*, 1, n)`.</span>
+<span class="sd">                              :math:`(*, 1, n)`, which will be broadcasted</span>
+<span class="sd">                              to the size of A in this function.</span>
 
 <span class="sd">    References::</span>
 <span class="sd">        - Nathan Halko, Per-Gunnar Martinsson, and Joel Tropp, Finding</span>
@@ -631,6 +632,7 @@ <h1>Source code for torch._lowrank</h1><div class="highlight"><pre>
     <span class="k">if</span> <span class="n">M</span> <span class="ow">is</span> <span class="kc">None</span><span class="p">:</span>
         <span class="n">M_t</span> <span class="o">=</span> <span class="kc">None</span>
     <span class="k">else</span><span class="p">:</span>
+        <span class="n">M</span> <span class="o">=</span> <span class="n">M</span><span class="o">.</span><span class="n">broadcast_to</span><span class="p">(</span><span class="n">A</span><span class="o">.</span><span class="n">size</span><span class="p">())</span>
         <span class="n">M_t</span> <span class="o">=</span> <span class="n">_utils</span><span class="o">.</span><span class="n">transpose</span><span class="p">(</span><span class="n">M</span><span class="p">)</span>
     <span class="n">A_t</span> <span class="o">=</span> <span class="n">_utils</span><span class="o">.</span><span class="n">transpose</span><span class="p">(</span><span class="n">A</span><span class="p">)</span>
 

docs/main/_modules/torch/_lowrank.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_tensor.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/_tensor_str.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 
@@ -482,11 +482,14 @@
   <h1>Source code for torch._utils</h1><div class="highlight"><pre>
 <span></span><span class="kn">import</span> <span class="nn">copyreg</span>
 <span class="kn">import</span> <span class="nn">functools</span>
+<span class="kn">import</span> <span class="nn">logging</span>
 <span class="kn">import</span> <span class="nn">sys</span>
 <span class="kn">import</span> <span class="nn">traceback</span>
 <span class="kn">import</span> <span class="nn">warnings</span>
 <span class="kn">from</span> <span class="nn">collections</span> <span class="kn">import</span> <span class="n">defaultdict</span>
-<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Any</span><span class="p">,</span> <span class="n">DefaultDict</span><span class="p">,</span> <span class="n">List</span><span class="p">,</span> <span class="n">Optional</span>
+<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">Any</span><span class="p">,</span> <span class="n">Callable</span><span class="p">,</span> <span class="n">DefaultDict</span><span class="p">,</span> <span class="n">Generic</span><span class="p">,</span> <span class="n">List</span><span class="p">,</span> <span class="n">Optional</span>
+
+<span class="kn">from</span> <span class="nn">typing_extensions</span> <span class="kn">import</span> <span class="n">ParamSpec</span>
 
 <span class="kn">import</span> <span class="nn">torch</span>
 
@@ -1417,6 +1420,28 @@ <h1>Source code for torch._utils</h1><div class="highlight"><pre>
 
     <span class="k">def</span> <span class="nf">get_calls</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">List</span><span class="p">:</span>
         <span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">call_order</span>
+
+
+<span class="n">logger</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">getLogger</span><span class="p">(</span><span class="vm">__name__</span><span class="p">)</span>
+<span class="n">P</span> <span class="o">=</span> <span class="n">ParamSpec</span><span class="p">(</span><span class="s2">&quot;P&quot;</span><span class="p">)</span>
+
+
+<span class="k">class</span> <span class="nc">CallbackRegistry</span><span class="p">(</span><span class="n">Generic</span><span class="p">[</span><span class="n">P</span><span class="p">]):</span>
+    <span class="k">def</span> <span class="fm">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">:</span> <span class="nb">str</span><span class="p">):</span>
+        <span class="bp">self</span><span class="o">.</span><span class="n">name</span> <span class="o">=</span> <span class="n">name</span>
+        <span class="bp">self</span><span class="o">.</span><span class="n">callback_list</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Callable</span><span class="p">[</span><span class="n">P</span><span class="p">,</span> <span class="kc">None</span><span class="p">]]</span> <span class="o">=</span> <span class="p">[]</span>
+
+    <span class="k">def</span> <span class="nf">add_callback</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">cb</span><span class="p">:</span> <span class="n">Callable</span><span class="p">[</span><span class="n">P</span><span class="p">,</span> <span class="kc">None</span><span class="p">])</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
+        <span class="bp">self</span><span class="o">.</span><span class="n">callback_list</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">cb</span><span class="p">)</span>
+
+    <span class="k">def</span> <span class="nf">fire_callbacks</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="o">*</span><span class="n">args</span><span class="p">:</span> <span class="n">P</span><span class="o">.</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">:</span> <span class="n">P</span><span class="o">.</span><span class="n">kwargs</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
+        <span class="k">for</span> <span class="n">cb</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">callback_list</span><span class="p">:</span>
+            <span class="k">try</span><span class="p">:</span>
+                <span class="n">cb</span><span class="p">(</span><span class="o">*</span><span class="n">args</span><span class="p">,</span> <span class="o">**</span><span class="n">kwargs</span><span class="p">)</span>
+            <span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
+                <span class="n">logger</span><span class="o">.</span><span class="n">exception</span><span class="p">(</span>
+                    <span class="s2">&quot;Exception in callback for </span><span class="si">%s</span><span class="s2"> registered with gpu trace&quot;</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">name</span>
+                <span class="p">)</span>
 </pre></div>
 
              </article>

docs/main/_modules/torch/_utils.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/amp/autocast_mode.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/modules/fused.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/qat/modules/conv_fused.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/qat/modules/linear_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/dynamic/modules/linear_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/modules/bn_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/modules/conv_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/intrinsic/quantized/modules/linear_relu.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/qat/dynamic/modules/linear.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/qat/modules/conv.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/qat/modules/linear.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantizable/modules/activation.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantizable/modules/rnn.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/dynamic/modules/linear.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/dynamic/modules/rnn.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/functional.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/activation.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/batchnorm.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/conv.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/embedding_ops.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/functional_modules.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/linear.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/nn/quantized/modules/normalization.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/ns/_numeric_suite.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>
 
 

docs/main/_modules/torch/ao/ns/_numeric_suite_fx.html

@@ -242,7 +242,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git6b8205d ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.4.0a0+git2b1ba0c ) &#x25BC</a>
     </div>