「 AI 」 一月 11, 2026
torch.nn.Embedding 中 max_norm 的作用
文章字数 7.7k 阅读约需 7 mins.
https://docs.pytorch.org/docs/stable/generated/torch.nn.Embedding.html#torch.nn.Embedding
nn.Embedding(num_embeddings, embedding_dim) 可以看成是一个查表词典:
- 有
num_embeddings行,每一行是一个embedding_dim维的向量。 - 输入是索引(比如单词 ID、类别 ID),输出是对应行的向量。
max_norm 的作用:
给这个“词典...
查看全文