pytorch
文章平均质量分 77
Leora_lh
这个作者很懒,什么都没留下…
展开
-
Pytorch里的CrossEntropyLoss详解
首先要知道上面提到的这些函数一部分是来自于torch.nn,而另一部分则来自于torch.nn.functional(常缩写为F)。二者函数的区别可参见知乎:torch.nn和funtional函数区别是什么? 下面是对与cross entropy有关的函数做的总结: torch.nn torch.nn.functional (F) CrossEntropyLoss cross_entropy LogSoftmax log_softmax NLLLoss nll_l.转载 2021-09-17 11:43:47 · 4108 阅读 · 0 评论 -
torch.nn.Embedding
转自:https://blog.csdn.net/foneone/article/details/103881687 torch.nn.Embedding模块可以看做一个字典,字典中每个索引对应一个词和词的embedding形式。利用这个模块,可以给词做embedding的初始化操作。 模块中主要参数: num_embeddings :字典中词的个数 embedding_dim:embedding的维度 padding_idx(索引指定填充):如果给定,则遇到padding_idx中的索引,则..转载 2020-10-23 10:18:48 · 177 阅读 · 0 评论