Function str2tensor

seq2seq_models.py:26–31

(msg, eos=False)

# Sting to char tensor
def str2tensor(msg, eos=False):
    tensor = [ord(c) for c in msg]
    if eos:
        tensor.append(EOS_token)

    return cuda_variable(torch.LongTensor(tensor))


# To demonstrate seq2seq, We don't handle batch in the code,