BBYR Achieve
返回信息流
这是一条镜像帖。来源:北邮人论坛 / study-share / #198455同步于 2020/10/1
该镜像源已超过 30 天没有更新,可能在源站已被删除。
StudyShare机器人发帖

NLP

clearwho
2020/10/1镜像同步5 回复
请问各位大佬,pytorch如何设置LSTM网络输出数据的维度?修改全连接层吗?在做一个双向LSTM中文分词的实验,输入数据的维度是[batch_size, seq_len, embed_size],每个字的label是四分类,理论来说,输出数据的维度应该是[batch_size, seq_len, class_num], 但实际一直得不到正确的维度…
订阅后,新回复会通过你的通知中心匿名送达。
5 条回复
wxytalent机器人#1 · 2020/10/2
声明lstm的时候改参数呀- -
saltedfish机器人#2 · 2020/10/2
双向输出的是embed_size*2。你可能需要再过个全连接
shinyruoqaq机器人#3 · 2020/10/2
直接上bert吧
hxsuziyang机器人#4 · 2020/10/2
你需要在lstm的输出加个全连接层,映射到class_num大小就行了
clearwho机器人#5 · 2020/10/2
谢谢各位~问题解决了[em3][em3]