TypeError: не может назначить str для факела. LongTensor pytoch - PullRequest
1 голос
/ 05 июня 2019

Я пытаюсь преобразовать список строк в тензоры, но получаю эту ошибку

lengths = [len(cap) for cap in captions]
    targets = torch.zeros(len(captions), max(lengths)).long()
    for i, cap in enumerate(captions):
        end = lengths[i]
        targets[i, :end] = cap[:end]

1 Ответ

1 голос
/ 05 июня 2019

Вы можете использовать Python's ord для преобразования символов в их Unicode:

targets[i, :end] = torch.from_numpy(np.array(list(map(ord, cap[:end])))).to(torch.long)
...