Python PyTorch Vectors.get_vecs_by_tokens用法及代码示例

本文简要介绍python语言中 torchtext.vocab.Vectors.get_vecs_by_tokens 的用法。

用法:

get_vecs_by_tokens(tokens, lower_case_backup=False)

参数：

• tokens-令牌或令牌列表。如果 tokens 是字符串，则返回形状为 self.dim 的一维张量；如果 tokens 是字符串列表，则返回 shape=(len(tokens), self.dim) 的二维张量。

• lower_case_backup-是否查找小写的 token。如果为 False，则将查找原始情况下的每个标记；如果为 True，则将首先查找原始大小写的每个标记，如果在属性 stoi 的键中找不到，则将查找小写的标记。默认值：假。

查找标记的嵌入向量。

例子

>>> examples = ['chip', 'baby', 'Beautiful']
>>> vec = text.vocab.GloVe(name='6B', dim=50)
>>> ret = vec.get_vecs_by_tokens(examples, lower_case_backup=True)