DotProduct.py 文件源码

def updateOutput(self, input):
        input1, input2 = input[0], input[1]

        if not self.buffer:
           self.buffer = input1.new()

        torch.mul(self.buffer, input1, input2)
        torch.sum(self.output, self.buffer, 1)
        self.output.resize_(input1.size(0))
        return self.output