How to extract typical rows in MXNet?
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]])
How to get the selected data? I tried the Pick and take functions, but don't know how to do it.
Solution
It seems gather_nd works
mx.nd.gather_nd(data,mx.nd.array([[0,0,1,1],[0,1,1,2]])).reshape(2,2,4)
- Static graph is fast. Dynamic graph is slow. Is there any specific benchmark demonstrating this?
- How is log_softmax() implemented to compute its value (and gradient) with better speed and numerical stability?
- Adding <NDArray 4 @gpu(0)> and regular np.array
- In mxnet, how is the derivative of 2x^2 equal to 4x when x is a matrix?
- RuntimeError: Cannot find the MXNet library
- How do I denormalize L2-Normalized Data in MXNet?
- Error when trying to train Deep Java Library neural Network with an Image folder Dataset
- The mxnet's cpp package reports error when #include regarding dmlc/base.h
- Python multiprocessing.Pool cannot pickle mxnet.mod.Module object
- Is there a way to check if mxnet uses my gpu?
- How to move entire net parameters from gpu to cpu?
- MXNET - How to add dropout layer to ResNet_v1 pretrained model
- How to convert mxnet.ndarray.ndarray.NDArray to a scalar?
- Sagemaker Serverless Inference & custom container: Model archiver subprocess fails
- How to install mxnet on google colab?
- Getting OSError: libnccl.so.2 while importing mxnet
- How to call a custom mxnet operator in DJL (Deep Java Library)?
- Sudden drop in accuracy while training a deep neural net
- Cannot import mxnet package in my environment
- Running GluonCV object detection model on Android
- Dlr model gives notorious result for object detection model
- Mxnet, load data from csv file with pandas and feed to NN models
- Issue when Re-implement Matrix Factorization in Pytorch
- How to extract typical rows in MXNet?
- Reshaping OpenCV Image (numpy) Dimensions
- mxnet (gluon): cpu used when gpu(0) context selected
- RecordIO: "The header of the MXNet RecordIO record...does not start with a valid magic number"
- gluon Sequential model: params return empty collection
- How to use the func like torch.nn.functional.conv2d() in mxnet?
- How to find memory leak in Python MXNet?