PairwiseDistance.py 文件源码

def updateOutput(self, input):
        self.output.resize_(1)
        assert input[0].dim() == 2

        self.diff = self.diff or input[0].new()

        torch.add(self.diff, input[0], -1, input[1]).abs_()

        self.output.resize_(input[0].size(0))
        self.output.zero_()
        self.output.add_(self.diff.pow_(self.norm).sum(1))
        self.output.pow_(1./self.norm)

        return self.output