Operation for nested tensors

Johannes_Vogt: <p>Hey, I appreciate the concept of nested tensors very much, but don’t understand which operations are already available for them.</p>
<p>While it seems to work with dropout, it does not with sigmoid. Where do I find the list of available operations?</p>
<p>In the end, I would like to apply an operation, such as  sorting, to many tensors of different size as good as possible in parallel.</p>

soulitzer: <p>There’s a script that can generate can a report of the current testing status of NJT ops that are tested with the latest NJT testing infra.</p><aside class="onebox githubpullrequest" data-onebox-src="https://github.com/pytorch/pytorch/pull/143311">
  <header class="source">

      <a href="https://github.com/pytorch/pytorch/pull/143311" target="_blank" rel="noopener nofollow ugc">github.com/pytorch/pytorch</a>
  </header>

  <article class="onebox-body">
    <div class="github-row" data-github-private-repo="false">



    <div class="github-icon-container" title="Pull Request">
      <svg width="60" height="60" class="github-icon" viewBox="0 0 12 16" aria-hidden="true"><path fill-rule="evenodd" d="M11 11.28V5c-.03-.78-.34-1.47-.94-2.06C9.46 2.35 8.78 2.03 8 2H7V0L4 3l3 3V4h1c.27.02.48.11.69.31.21.2.3.42.31.69v6.28A1.993 1.993 0 0 0 10 15a1.993 1.993 0 0 0 1-3.72zm-1 2.92c-.66 0-1.2-.55-1.2-1.2 0-.65.55-1.2 1.2-1.2.65 0 1.2.55 1.2 1.2 0 .65-.55 1.2-1.2 1.2zM4 3c0-1.11-.89-2-2-2a1.993 1.993 0 0 0-1 3.72v6.56A1.993 1.993 0 0 0 2 15a1.993 1.993 0 0 0 1-3.72V4.72c.59-.34 1-.98 1-1.72zm-.8 10c0 .66-.55 1.2-1.2 1.2-.65 0-1.2-.55-1.2-1.2 0-.65.55-1.2 1.2-1.2.65 0 1.2.55 1.2 1.2zM2 4.2C1.34 4.2.8 3.65.8 3c0-.65.55-1.2 1.2-1.2.65 0 1.2.55 1.2 1.2 0 .65-.55 1.2-1.2 1.2z"></path></svg>
    </div>

  <div class="github-info-container">



      <h4>
        <a href="https://github.com/pytorch/pytorch/pull/143311" target="_blank" rel="noopener nofollow ugc">Script to generate NJT OpInfo testing report</a>
      </h4>

    <div class="branches">
      <code>gh/jbschlosser/214/base</code> ← <code>gh/jbschlosser/214/head</code>
    </div>

      <div class="github-info">
        <div class="date">
          opened <span class="discourse-local-date" data-format="ll" data-date="2024-12-16" data-time="19:18:38" data-timezone="UTC">07:18PM - 16 Dec 24 UTC</span>
        </div>

        <div class="user">
          <a href="https://github.com/jbschlosser" target="_blank" rel="noopener nofollow ugc">
            <img alt="" src="https://avatars.githubusercontent.com/u/75754324?v=4" class="onebox-avatar-inline" width="20" height="20">
            jbschlosser
          </a>
        </div>

        <div class="lines" title="11 commits changed 1 files with 401 additions and 0 deletions">
          <a href="https://github.com/pytorch/pytorch/pull/143311/files" target="_blank" rel="noopener nofollow ugc">
            <span class="added">+401</span>
            <span class="removed">-0</span>
          </a>
        </div>
      </div>
  </div>
</div>

  <div class="github-row">
    <p class="github-body-container">Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom):
* __<span class="show-more-container"><a href="https://github.com/pytorch/pytorch/pull/143311" target="_blank" rel="noopener nofollow ugc" class="show-more">…</a></span><span class="excerpt hidden">-&gt;__ #143311
* #142063

I hacked together a script to parse skips / xfails for the OpInfo-based tests and visualize the status of passing tests in a table. Results are reported op-wise for `(forward, compile_forward, backward, compile_backward) x (contiguous, non-contiguous)`, including the (color-coded) success rate / number of xfails / number of skips.

![njt_op_report](https://github.com/user-attachments/assets/859463f7-2399-44cd-9144-cc28d39b0524)

~~TODO: It might be interesting to have a collapsible xfail / skip explorer?~~ Added this

Yes I know it looks terrible! I'm not a frontend dev and it's more important to visualize the data so I can prioritize fixes than spend time making it look nice.

TBD how to incorporate this into the NJT docs.</span></p>
  </div>

  </article>

  <div class="onebox-metadata">
    
    
  </div>

  <div style="clear: both"></div>
</aside>

<p>(Although I’m not sure all ops supported is necessarily tested with this latest testing infra.)</p>