Sparse layers 字面意思就是稀疏层,我是查torch.nn.Embedding看到的
torch.nn.Embedding(num_embeddings, embedding_dim, padding_idx=None, max_norm=None, norm_type=2, scale_grad_by_freq=False, sparse=False)
num_embeddings (int) - 嵌入字典的大小
embedding_dim (int) - 每个嵌入向量的大小
其他参数见文档
变量:
weight (Tensor) -形状为(num_embeddings, embedding_dim)的模块中可学习的权值