How to extract typical rows in MXNet?

Question

These are data(Batch size 2) and batch index

import mxnet as mx
data=mx.nd.array(range(24)).reshape(2,3,4)
index=mx.nd.array([[0,1],[1,2]])

<