Add training version of RnnEmb