NPX: NumPy Neural Network Extension =================================== .. currentmodule:: mxnet.npx Compatibility ------------- .. autosummary:: :toctree: generated/ set_np reset_np .. code:: is_np_array use_np_array is_np_shape use_np_shape np_array np_shape Devices --------- .. autosummary:: :toctree: generated/ cpu cpu_pinned gpu gpu_memory_info current_context num_gpus Nerual networks ----------------------- .. autosummary:: :toctree: generated/ activation batch_norm convolution dropout embedding fully_connected layer_norm pooling rnn leaky_relu multibox_detection multibox_prior multibox_target roi_pooling More operators ------------------ .. autosummary:: :toctree: generated/ sigmoid smooth_l1 softmax threading topk waitall load save one_hot pick reshape_like batch_flatten batch_dot gamma sequence_mask .. code:: seed