这个屌丝很懒，什么也没留下！

pytorch 笔记： torch.nn.Embedding_pytorch embeding的权重

作者：盐析白兔 | 2024-04-05 06:15:27

pytorch embeding的权重

pytorch中，针对词向量有一个专门的层nn.Embedding，用来实现词与词向量的映射。

nn.Embedding具有一个权重，形状是(num_embeddings,embedding_dimension)。

例如，输入10个词，每个词用2维向量表征，对应的权重就是一个10x2的矩阵。

如果Embedding层的输入形状为NxM（比如N是batch size,M是序列的长度），则输出的形状是N*M*embedding_dimension.

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/363853