[Relax] Allow PrimValue as index in relax.op.take#16940

Lunderberg merged 3 commits intoapache:mainapache/tvm:mainfrom

Lunderberg:relax_prim_value_as_take_indexLunderberg/tvm:relax_prim_value_as_take_indexCopy head branch name to clipboard