关于last
来源:7-14 计算图实现
CrazyCat岩
2019-01-06
请问last=rnn_outputs[:,-1,:]中为什么第二维是-1,没明白last最后是什么样子的。
写回答
1回答
-
rnn_outputs 的shape是[batch_size, num_timesteps, lstm_outputs[-1]], 在文本分类中,我们需要的是lstm最后一步的输出,所以在num_timesteps这一维取最后一个,所以last的维度是[batch_size, 1, lstm_outputs[-1]], 这时你可以调用reshape把中间那一维给去掉。
112019-02-13
相似问题