Add docs from pypose/pypose

Author: pyposebot

docs/preview/pr/352/_modules/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/basics/ops/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/func/jac/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/function/checking/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/function/geometry/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/function/linalg/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/function/spline/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/lietensor/basics/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 

docs/preview/pr/352/_modules/pypose/lietensor/convert/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 
@@ -241,8 +241,10 @@
 
   <h1>Source code for pypose.lietensor.convert</h1><div class="highlight"><pre>
 <span></span><span class="kn">import</span> <span class="nn">torch</span><span class="o">,</span> <span class="nn">warnings</span>
+<span class="kn">from</span> <span class="nn">torch.nn.functional</span> <span class="kn">import</span> <span class="n">normalize</span>
 <span class="kn">from</span> <span class="nn">.utils</span> <span class="kn">import</span> <span class="n">SO3</span><span class="p">,</span> <span class="n">so3</span><span class="p">,</span> <span class="n">SE3</span><span class="p">,</span> <span class="n">RxSO3</span><span class="p">,</span> <span class="n">Sim3</span>
-<span class="kn">from</span> <span class="nn">.lietensor</span> <span class="kn">import</span> <span class="n">LieTensor</span><span class="p">,</span> <span class="n">SE3_type</span><span class="p">,</span> <span class="n">SO3_type</span><span class="p">,</span> <span class="n">Sim3_type</span><span class="p">,</span> <span class="n">RxSO3_type</span>
+<span class="kn">from</span> <span class="nn">.lietensor</span> <span class="kn">import</span> <span class="n">LieTensor</span><span class="p">,</span>  <span class="n">liegroup</span><span class="p">,</span> <span class="n">liealgebra</span>
+<span class="kn">from</span> <span class="nn">.lietensor</span> <span class="kn">import</span> <span class="n">SE3_type</span><span class="p">,</span> <span class="n">SO3_type</span><span class="p">,</span> <span class="n">Sim3_type</span><span class="p">,</span> <span class="n">RxSO3_type</span>
 
 
 <div class="viewcode-block" id="mat2SO3"><a class="viewcode-back" href="../../../../generated/pypose.mat2SO3/#pypose.mat2SO3">[docs]</a><span class="k">def</span> <span class="nf">mat2SO3</span><span class="p">(</span><span class="n">mat</span><span class="p">,</span> <span class="n">check</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">rtol</span><span class="o">=</span><span class="mf">1e-5</span><span class="p">,</span> <span class="n">atol</span><span class="o">=</span><span class="mf">1e-5</span><span class="p">):</span>
@@ -1066,6 +1068,41 @@ <h1>Source code for pypose.lietensor.convert</h1><div class="highlight"><pre>
 <span class="sd">    See :obj:`euler2SO3` for more information.</span>
 <span class="sd">    &#39;&#39;&#39;</span>
     <span class="k">return</span> <span class="n">inputs</span><span class="o">.</span><span class="n">euler</span><span class="p">(</span><span class="n">eps</span><span class="o">=</span><span class="n">eps</span><span class="p">)</span></div>
+
+
+<div class="viewcode-block" id="quat2unit"><a class="viewcode-back" href="../../../../generated/pypose.quat2unit/#pypose.quat2unit">[docs]</a><span class="k">def</span> <span class="nf">quat2unit</span><span class="p">(</span><span class="nb">input</span><span class="p">:</span> <span class="n">LieTensor</span><span class="p">,</span> <span class="n">eps</span><span class="o">=</span><span class="mf">1e-12</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="n">LieTensor</span><span class="p">:</span>
+<span class="w">    </span><span class="sa">r</span><span class="sd">&#39;&#39;&#39;</span>
+<span class="sd">    Normalize the quaternion part of a ``LieTensor``, which has to be a Lie group.</span>
+<span class="sd">    If input is a not a Lie group, then do nothing and return the input.</span>
+<span class="sd">    If the quaternion parts contain pure zeros, then raise an error.</span>
+
+<span class="sd">    The quaternion parts :math:`v` are normalized as</span>
+
+<span class="sd">    .. math::</span>
+<span class="sd">        v = \frac{v}{\max(\lVert v \rVert_2, \epsilon)},</span>
+
+<span class="sd">    where :math:`\epsilon` is a small value to avoid division by zero</span>
+
+<span class="sd">    Args:</span>
+<span class="sd">        input (``LieTensor``): input LieTensor of any type and shape.</span>
+<span class="sd">        eps (``float``): small value to avoid division by zero. Default: 1e-12.</span>
+
+<span class="sd">    Return:</span>
+<span class="sd">        :obj:`LieTensor`: the output LieTensor.</span>
+<span class="sd">    &#39;&#39;&#39;</span>
+    <span class="k">if</span> <span class="nb">isinstance</span><span class="p">(</span><span class="nb">input</span><span class="p">,</span> <span class="n">LieTensor</span><span class="p">)</span> <span class="ow">and</span> <span class="p">(</span><span class="nb">input</span><span class="o">.</span><span class="n">ltype</span> <span class="ow">in</span> <span class="n">liegroup</span><span class="p">):</span>
+        <span class="n">data</span> <span class="o">=</span> <span class="nb">input</span><span class="o">.</span><span class="n">tensor</span><span class="p">()</span>
+        <span class="k">if</span> <span class="nb">input</span><span class="o">.</span><span class="n">ltype</span> <span class="ow">in</span> <span class="p">[</span><span class="n">SO3_type</span><span class="p">,</span> <span class="n">RxSO3_type</span><span class="p">]:</span>
+            <span class="n">data</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="p">:</span><span class="mi">4</span><span class="p">]</span> <span class="o">=</span> <span class="n">normalize</span><span class="p">(</span><span class="n">data</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="p">:</span><span class="mi">4</span><span class="p">],</span> <span class="n">p</span><span class="o">=</span><span class="mi">2</span><span class="p">,</span> <span class="n">dim</span><span class="o">=-</span><span class="mi">1</span><span class="p">,</span> <span class="n">eps</span><span class="o">=</span><span class="n">eps</span><span class="p">)</span>
+        <span class="k">elif</span> <span class="nb">input</span><span class="o">.</span><span class="n">ltype</span> <span class="ow">in</span> <span class="p">[</span><span class="n">SE3_type</span><span class="p">,</span> <span class="n">Sim3_type</span><span class="p">]:</span>
+            <span class="n">data</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="mi">3</span><span class="p">:</span><span class="mi">7</span><span class="p">]</span> <span class="o">=</span> <span class="n">normalize</span><span class="p">(</span><span class="n">data</span><span class="p">[</span><span class="o">...</span><span class="p">,</span> <span class="mi">3</span><span class="p">:</span><span class="mi">7</span><span class="p">],</span> <span class="n">p</span><span class="o">=</span><span class="mi">2</span><span class="p">,</span> <span class="n">dim</span><span class="o">=-</span><span class="mi">1</span><span class="p">,</span> <span class="n">eps</span><span class="o">=</span><span class="n">eps</span><span class="p">)</span>
+        <span class="n">output</span> <span class="o">=</span> <span class="n">LieTensor</span><span class="p">(</span><span class="n">data</span><span class="p">,</span> <span class="n">ltype</span><span class="o">=</span><span class="nb">input</span><span class="o">.</span><span class="n">ltype</span><span class="p">)</span>
+        <span class="k">if</span> <span class="p">(</span><span class="n">output</span><span class="o">.</span><span class="n">rotation</span><span class="p">()</span><span class="o">.</span><span class="n">norm</span><span class="p">(</span><span class="n">p</span><span class="o">=</span><span class="mi">2</span><span class="p">,</span> <span class="n">dim</span><span class="o">=-</span><span class="mi">1</span><span class="p">)</span> <span class="o">&lt;</span> <span class="n">eps</span><span class="p">)</span><span class="o">.</span><span class="n">any</span><span class="p">():</span>
+            <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Detected zero quaternions, which cannot be normalized.&quot;</span><span class="p">)</span>
+        <span class="k">return</span> <span class="n">output</span>
+    <span class="k">else</span><span class="p">:</span>
+        <span class="n">warnings</span><span class="o">.</span><span class="n">warn</span><span class="p">(</span><span class="s2">&quot;Input is not Lie group, doing thing and returning input..&quot;</span><span class="p">)</span>
+        <span class="k">return</span> <span class="nb">input</span></div>
 </pre></div>
 
              </article>

docs/preview/pr/352/_modules/pypose/lietensor/lietensor/index.html

@@ -125,7 +125,7 @@
           <div class="pytorch-left-menu-search">
 
     <div class="version">
-      <a href='https://pypose.org/docs/versions.html'>0.6.8 &#x25BC</a>
+      <a href='https://pypose.org/docs/versions.html'>0.6.9 &#x25BC</a>
     </div>
 
 
@@ -982,7 +982,8 @@ <h1>Source code for pypose.lietensor.lietensor</h1><div class="highlight"><pre>
 <span class="n">SE3_type</span><span class="p">,</span> <span class="n">se3_type</span> <span class="o">=</span> <span class="n">SE3Type</span><span class="p">(),</span> <span class="n">se3Type</span><span class="p">()</span>
 <span class="n">Sim3_type</span><span class="p">,</span> <span class="n">sim3_type</span> <span class="o">=</span> <span class="n">Sim3Type</span><span class="p">(),</span> <span class="n">sim3Type</span><span class="p">()</span>
 <span class="n">RxSO3_type</span><span class="p">,</span> <span class="n">rxso3_type</span> <span class="o">=</span> <span class="n">RxSO3Type</span><span class="p">(),</span> <span class="n">rxso3Type</span><span class="p">()</span>
-
+<span class="n">liegroup</span> <span class="o">=</span> <span class="p">[</span><span class="n">SO3_type</span><span class="p">,</span> <span class="n">SE3_type</span><span class="p">,</span> <span class="n">Sim3_type</span><span class="p">,</span> <span class="n">RxSO3_type</span><span class="p">]</span>
+<span class="n">liealgebra</span> <span class="o">=</span> <span class="p">[</span><span class="n">so3_type</span><span class="p">,</span> <span class="n">se3_type</span><span class="p">,</span> <span class="n">sim3_type</span><span class="p">,</span> <span class="n">rxso3_type</span><span class="p">]</span>
 
 <div class="viewcode-block" id="LieTensor"><a class="viewcode-back" href="../../../../generated/pypose.LieTensor/#pypose.LieTensor">[docs]</a><span class="k">class</span> <span class="nc">LieTensor</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="w">    </span><span class="sa">r</span><span class="sd">&quot;&quot;&quot; A sub-class of :obj:`torch.Tensor` to represent Lie Algebra and Lie Group.</span>