TypeError: only integer scalar arrays can be converted to a scalar index about seq2seq-couplet

[LongHuW]

我这边是因为labels出现了三维array这种数据，所以我加了一个步骤，将其降维成一维就好了

def decode_text(labels, vocabs, end_token = '</s>'):
    results = []
    if isinstance(labels, list) == False and labels.ndim>1:
        labels_one = labels.reshape(-1).astype(int)
        for idx in labels_one:
            word = vocabs[idx]
            if word == end_token:
                return ' '.join(results)
            results.append(word)
        return ' '.join(results)

    for idx in labels:
        word = vocabs[idx]
        if word == end_token:
            return ' '.join(results)
        results.append(word)
    return ' '.join(results)

[xiaoshuai1]

同问

[zhoujinyang1996]

这是来自QQ邮箱的假期自动回复邮件。   您好，我最近正在休假中，无法亲自回复您的邮件。我将在假期结束后，尽快给您回复。

[Lijiaxing-lx]

您好，邮件已收到。我将尽快给你回复。

[Drizzlenum]

请问跑通了吗

[cui-z]

word = vocabs[idx] 修改为
word = np.array(vocabs)[idx]

[DaemonicCao]

感谢🙏 发自我的iPhone

…

在 2022年5月4日，上午11:16，zhoujinyang1996 ***@***.***> 写道：  这是来自QQ邮箱的假期自动回复邮件。 您好，我最近正在休假中，无法亲自回复您的邮件。我将在假期结束后，尽快给您回复。 — Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.