search

131250208 / TPlinker-joint-extraction

438 stars 94 forks source link

seq_len = seq_hiddens.size()[-2] #76

Open hjing100 opened 2 years ago

hjing100 commented 2 years ago

HandshakingKernel类中的seq_len = seq_hiddens.size()[-2]是不是不太对， 因为是seq_hiddens: (batch_size, seq_len, hidden_size)， seq_len = seq_hiddens.size()[-1]好像才对？

131250208 commented 2 years ago

@hjing100 -1取的是最后一个hidden_size