Pytorch cnn nlp