This cheatsheet serves as a quick reference for PyTorch users who are interested in trying MXNet, and vice versa.
Pytorch is a deep learning framework provides imperative tensor manipulation and neural network training. MXNet provides similar imperative tensor manipulation through the ndarray
package and neural network training through gluon
. This cheatsheet maps functions one-by-one between these two frameworks.
Note that MXNet has a symbolic interface similar to Keras and Tensorflow that may provide better performance and portability. This cheatsheet mainly focus on MXNet's imperative interface.