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. ---------------------------------------------------------------- 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: [email protected]
