Imported from https://github.com/ChaiBapchya/mxnet-hardmax_machine_translation