torch中数组使用技巧

两个数组合并成一个数组

src_vocab[src_sentence.lower().split(' ')] + [
        src_vocab['<eos>']]

tensor中添加数组然后再增加第一个维度

torch.unsqueeze(torch.tensor(
        [tgt_vocab['<bos>']], dtype=torch.long, device=device), dim=0)

奇数位与偶数位添加函数

self.P[:, :, 0::2] = torch.sin(X)
self.P[:, :, 1::2] = torch.cos(X)

定义一个序列化，然后添加序列模型，可以使用for循环然后结合add_module添加

y_hat是个数组，根据数组维度选择对应维度最大的一个数字

y_hat.argmax(axis=1)

数组中的遍历

ret_tokens = []
        for line in tokens:
            for token in line:
                ret_tokens.append(token)

# 改方法的做法等同于上面的遍历形式
tokens = [token for line in tokens for token in line]