Textcnn中的卷积维度的计算,以及经过textcnn维度如何处理:
输入词向量
维度: batch_size sen_len embedding_dim
卷积:
维度: channel kernel_size
卷积后的维度:
假如输入维度为 :128,12,128,卷积核的大小:6,3,128
需要unsquenee 增加一维 128 1 12 128
维度:128 6 10 1
去掉第三维 128 6 10
max池化:
以输入.size(2)为池化的大小:
维度: 128 6 1
去掉第二维:128 6
经过 全连接层:128 6 10 假设有10个分类
最后 实现分类
2021-04-05
最新推荐文章于 2022-11-20 23:38:10 发布