pytorch里的nn.Embedding是什么东西?

pytorch里的nn.Embedding是什么东西?

假设有n个数据,每个数据对应一个d维特征,那么可以用nn.Embedding(n, d)来保存这些特征,这个embedding接收索引,然后把对应位置的特征返回给调用者

当然它也可以用来保存类中心向量

你可能感兴趣的:(DL,Python,pytorch,深度学习,机器学习,nlp,embedding)