modified from https://github.com/chenhongshen/NNHetSeq - View it on GitHub