随笔分类 -  pytorch

摘要:而RNN本质上就是linear layers。 即使RNN的输入数据是batch first,内部也会转为seq_len first。 def forward(self, input, hx=None): batch_sizes = None # is not packed, batch_sizes 阅读全文
posted @ 2019-12-25 18:14 甲乙丙2333 阅读(621) 评论(0) 推荐(0)