Hi i think nested tensors were implemented in DETR. To get the shape, use: Nested_Tensor.tensors.shape
Here’s the link to the DETR hands-on where I found it; its in the attention visualisation section. Cheers!
  
      

      colab.research.google.com
  

  
    

Google Colaboratory 24