TPlinker-joint-extraction icon indicating copy to clipboard operation
TPlinker-joint-extraction copied to clipboard

seq_len = seq_hiddens.size()[-2]

Open hjing100 opened this issue 2 years ago • 1 comments

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

hjing100 avatar Jun 13 '22 12:06 hjing100

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

131250208 avatar Jun 27 '22 03:06 131250208