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)