摘要:
1. torch.nn.Embedding embedding 的主要目的是对稀疏特征进行降维。函数原型如下: """ num_embeddings – 词典的大小 embedding_dim – 嵌入向量的维度,即用多少维的向量来表示一个词 padding_idx(optional) – 填充id 阅读全文
摘要:
nn.ModuleList 可以把任意 nn.Module 的子类 (比如 nn.Conv2d, nn.Linear 之类的) 加到这个 list 里面,方法和 Python 自带的 list 一样,无非 是 extend,append 等操作。但不同于一般的 list,加入到 nn.ModuleL 阅读全文