We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
af49870
einx.rearrange
einx.{elementwise|add|multiply|where|...}
einx.{reduce|sum|mean|...}
einx.{vmap_with_axis|flip|softmax|...}
einx.dot
einx.vmap
vmap
einx.{index|get_at|set_at|...}
tf.gather_nd
x[y]
einx.get_at