You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/02/03 02:30:38 UTC

[GitHub] [tvm] codeislife99 edited a comment on pull request #7373: Add Support for aten::index

codeislife99 edited a comment on pull request #7373:
URL: https://github.com/apache/tvm/pull/7373#issuecomment-770105376


   The behaviour is much different right from implementation to output shape size and unique to PyTorch. I was initially thinking of adding this into a new operator similar to aten::masked_select. However there are very slight differences in input and output shapes between `aten::masked_select` and `aten::index + boolean indices` as well , so I had to do this. 
   
   Correction: It seems that numpy also follows the same concepts. 


----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

For queries about this service, please contact Infrastructure at:
users@infra.apache.org