Imported from https://github.com/cpmpercussion/keras-mdn-layer