revert arm map in Softmax