Generate Python docs from pytorch/pytorch@e7f12b1

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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#x25BC</a>
     </div>
 
 
@@ -529,6 +529,7 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
     <span class="s1">&#39;set_warn_always&#39;</span><span class="p">,</span> <span class="s1">&#39;is_warn_always_enabled&#39;</span><span class="p">,</span> <span class="s1">&#39;SymInt&#39;</span><span class="p">,</span> <span class="s1">&#39;SymFloat&#39;</span><span class="p">,</span>
     <span class="s1">&#39;SymBool&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_not&#39;</span><span class="p">,</span> <span class="s1">&#39;unravel_index&#39;</span><span class="p">,</span>
     <span class="s1">&#39;sym_int&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_float&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_max&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_min&#39;</span><span class="p">,</span> <span class="s1">&#39;sym_ite&#39;</span><span class="p">,</span> <span class="s1">&#39;compile&#39;</span><span class="p">,</span> <span class="s1">&#39;vmap&#39;</span><span class="p">,</span>
+    <span class="s1">&#39;sym_sqrt&#39;</span><span class="p">,</span>
     <span class="s1">&#39;export&#39;</span><span class="p">,</span> <span class="s1">&#39;autocast&#39;</span><span class="p">,</span> <span class="s1">&#39;cond&#39;</span><span class="p">,</span>
 <span class="p">]</span>
 
@@ -887,8 +888,15 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
 <span class="sd">    Args:</span>
 <span class="sd">        a (SymBool or bool): Object to negate</span>
 <span class="sd">    &quot;&quot;&quot;</span>
+    <span class="kn">import</span> <span class="nn">sympy</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function_unary</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function_unary</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_not</span><span class="p">,</span> <span class="p">(</span><span class="n">a</span><span class="p">,),</span> <span class="n">a</span><span class="p">)</span>
     <span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="s1">&#39;__sym_not__&#39;</span><span class="p">):</span>
         <span class="k">return</span> <span class="n">a</span><span class="o">.</span><span class="n">__sym_not__</span><span class="p">()</span>
+    <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">sympy</span><span class="o">.</span><span class="n">Basic</span><span class="p">):</span>
+        <span class="k">return</span> <span class="o">~</span><span class="n">a</span>  <span class="c1"># type: ignore[operator]</span>
     <span class="k">return</span> <span class="ow">not</span> <span class="n">a</span></div>
 
 <div class="viewcode-block" id="sym_float"><a class="viewcode-back" href="../generated/torch.sym_float.html#torch.sym_float">[docs]</a><span class="k">def</span> <span class="nf">sym_float</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
@@ -897,6 +905,10 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
 <span class="sd">    Args:</span>
 <span class="sd">        a (SymInt, SymFloat, or object): Object to cast</span>
 <span class="sd">    &quot;&quot;&quot;</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function_unary</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function_unary</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_float</span><span class="p">,</span> <span class="p">(</span><span class="n">a</span><span class="p">,),</span> <span class="n">a</span><span class="p">)</span>
     <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">SymFloat</span><span class="p">):</span>
         <span class="k">return</span> <span class="n">a</span>
     <span class="k">elif</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="s1">&#39;__sym_float__&#39;</span><span class="p">):</span>
@@ -910,6 +922,10 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
 <span class="sd">    Args:</span>
 <span class="sd">        a (SymInt, SymFloat, or object): Object to cast</span>
 <span class="sd">    &quot;&quot;&quot;</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function_unary</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function_unary</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_int</span><span class="p">,</span> <span class="p">(</span><span class="n">a</span><span class="p">,),</span> <span class="n">a</span><span class="p">)</span>
     <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">SymInt</span><span class="p">):</span>
         <span class="k">return</span> <span class="n">a</span>
     <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">SymFloat</span><span class="p">):</span>
@@ -918,6 +934,10 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
 
 <div class="viewcode-block" id="sym_max"><a class="viewcode-back" href="../generated/torch.sym_max.html#torch.sym_max">[docs]</a><span class="k">def</span> <span class="nf">sym_max</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot; SymInt-aware utility for max().&quot;&quot;&quot;</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function</span><span class="p">((</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_max</span><span class="p">,</span> <span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">),</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)</span>
     <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="p">(</span><span class="n">SymInt</span><span class="p">,</span> <span class="n">SymFloat</span><span class="p">)):</span>
         <span class="k">return</span> <span class="n">a</span><span class="o">.</span><span class="n">__sym_max__</span><span class="p">(</span><span class="n">b</span><span class="p">)</span>
     <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="p">(</span><span class="n">SymInt</span><span class="p">,</span> <span class="n">SymFloat</span><span class="p">)):</span>
@@ -929,13 +949,31 @@ <h1>Source code for torch</h1><div class="highlight"><pre>
 
 <div class="viewcode-block" id="sym_min"><a class="viewcode-back" href="../generated/torch.sym_min.html#torch.sym_min">[docs]</a><span class="k">def</span> <span class="nf">sym_min</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">):</span>
 <span class="w">    </span><span class="sd">&quot;&quot;&quot; SymInt-aware utility for max().&quot;&quot;&quot;</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function</span><span class="p">((</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_min</span><span class="p">,</span> <span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">),</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)</span>
     <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="p">(</span><span class="n">SymInt</span><span class="p">,</span> <span class="n">SymFloat</span><span class="p">)):</span>
         <span class="k">return</span> <span class="n">a</span><span class="o">.</span><span class="n">__sym_min__</span><span class="p">(</span><span class="n">b</span><span class="p">)</span>
     <span class="k">elif</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="p">(</span><span class="n">SymInt</span><span class="p">,</span> <span class="n">SymFloat</span><span class="p">)):</span>
         <span class="k">return</span> <span class="n">b</span><span class="o">.</span><span class="n">__sym_min__</span><span class="p">(</span><span class="n">a</span><span class="p">)</span>
     <span class="k">return</span> <span class="n">builtins</span><span class="o">.</span><span class="n">min</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">)</span>  <span class="c1"># type: ignore[operator]</span></div>
 
+<span class="c1"># Drop in replacement for math.sqrt</span>
+<span class="k">def</span> <span class="nf">sym_sqrt</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function_unary</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function_unary</span><span class="p">(</span><span class="n">a</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_sqrt</span><span class="p">,</span> <span class="p">(</span><span class="n">a</span><span class="p">,),</span> <span class="n">a</span><span class="p">)</span>
+    <span class="k">if</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="s2">&quot;__sym_sqrt__&quot;</span><span class="p">):</span>
+        <span class="k">return</span> <span class="n">a</span><span class="o">.</span><span class="n">__sym_sqrt__</span><span class="p">()</span>
+    <span class="k">return</span> <span class="n">math</span><span class="o">.</span><span class="n">sqrt</span><span class="p">(</span><span class="n">a</span><span class="p">)</span>
+
 <div class="viewcode-block" id="sym_ite"><a class="viewcode-back" href="../generated/torch.sym_ite.html#torch.sym_ite">[docs]</a><span class="k">def</span> <span class="nf">sym_ite</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">):</span>
+    <span class="kn">from</span> <span class="nn">.overrides</span> <span class="kn">import</span> <span class="n">has_torch_function</span><span class="p">,</span> <span class="n">handle_torch_function</span>
+
+    <span class="k">if</span> <span class="n">has_torch_function</span><span class="p">((</span><span class="n">b</span><span class="p">,</span> <span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">)):</span>
+        <span class="k">return</span> <span class="n">handle_torch_function</span><span class="p">(</span><span class="n">sym_ite</span><span class="p">,</span> <span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">),</span> <span class="n">b</span><span class="p">,</span> <span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">)</span>
     <span class="k">assert</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="p">(</span><span class="n">SymBool</span><span class="p">,</span> <span class="n">builtins</span><span class="o">.</span><span class="n">bool</span><span class="p">))</span> <span class="ow">and</span> <span class="nb">type</span><span class="p">(</span><span class="n">t</span><span class="p">)</span> <span class="o">==</span> <span class="nb">type</span><span class="p">(</span><span class="n">f</span><span class="p">)</span>
     <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">b</span><span class="p">,</span> <span class="n">SymBool</span><span class="p">):</span>
         <span class="k">return</span> <span class="n">b</span><span class="o">.</span><span class="n">__sym_ite__</span><span class="p">(</span><span class="n">t</span><span class="p">,</span> <span class="n">f</span><span class="p">)</span>

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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#x25BC</a>
     </div>
 
 

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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#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.2.0a0+git5a96a42 ) &#x25BC</a>
+      <a href='https://pytorch.org/docs/versions.html'>main (2.2.0a0+gite7f12b1 ) &#x25BC</a>
     </div>