Open hjing100 opened 2 years ago
HandshakingKernel类中的seq_len = seq_hiddens.size()[-2]是不是不太对, 因为是seq_hiddens: (batch_size, seq_len, hidden_size), seq_len = seq_hiddens.size()[-1]好像才对?
@hjing100 -1取的是最后一个hidden_size
HandshakingKernel类中的seq_len = seq_hiddens.size()[-2]是不是不太对, 因为是seq_hiddens: (batch_size, seq_len, hidden_size), seq_len = seq_hiddens.size()[-1]好像才对?