python
Leora_lh
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
torch.nn.Embedding
转自:https://blog.youkuaiyun.com/foneone/article/details/103881687torch.nn.Embedding模块可以看做一个字典,字典中每个索引对应一个词和词的embedding形式。利用这个模块,可以给词做embedding的初始化操作。模块中主要参数:num_embeddings :字典中词的个数embedding_dim:embedding的维度padding_idx(索引指定填充):如果给定,则遇到padding_idx中的索引,则..转载 2020-10-23 10:18:48 · 276 阅读 · 0 评论 -
torch.nn.Conv2d()函数详解
转自:https://blog.youkuaiyun.com/m0_37586991/article/details/87855342import torchx = torch.randn(2,1,7,3)conv = torch.nn.Conv2d(1,8,(2,3))res = conv(x)print(res.shape) # shape = (2, 8, 6, 1)输入:x[batch_size,channels,height_1,width_1]batch_size 一个bat转载 2020-10-22 10:56:41 · 1146 阅读 · 0 评论
分享