Implement index_select using oneDNN

[knowledgaction]

I want to implement the index_select operation. Can it be implemented using oneDNN? If so, could you provide an example program? Thank you.

[shu1chen]

Hello @knowledgaction, oneDNN offers the binary_select operation, which performs a conditional elementwise select operation. However, this is different from the index_select operation.
oneDNN does not provide a direct equivalent to the index_select operation, and implementing index_select using oneDNN is not straightforward.
Could you please provide more details about your specific requirements? Specifically, what are the use cases in which you intend to implement index_select using oneDNN?