nn.Embedding()个人记录

2023-12-24 05:55:10

?维度

import torch.nn as nn
 
embedding = nn.Embedding(num_embeddings = 10, embedding_dim = 256)

nn.Embedding()随机产生一个权重矩阵weight，维度为（num_embeddings, embedding_dim）?

输入维度（batch_size, Seq_len）

输出维度（batch_size，Seq_len，embedding_dim）

举例

?

参考&转载：

pytorch复习笔记--nn.Embedding()的用法-CSDN博客?

文章来源:https://blog.csdn.net/qq_55736201/article/details/135176357
本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。