layers.py 文件源码

def forward(self, q_input, a_input):
        qw = torch.mm(q_input, self.W.view(self.input_size, -1)).view(-1, self.dim, self.input_size)
        qwa = torch.bmm(qw, torch.unsqueeze(a_input, 2))
        qa_vec = qwa.view(-1, self.dim)
        return qa_vec