<human>:  How to find all elements in a numpy 2-dimensional array that match a certain list?
<bot>:).reshape(A.shape), A, 0)

#Warning: this code generation is experimental. Please inspect the code for any bugs before executing.