PartialLinear.py 文件源码

def accGradParameters(self, input, gradOutput, scale=1):
        self.network.accGradParameters([input, self.partition], gradOutput, scale)
        if self.bias:
            self.buffer = self.buffer or input.new()
            self.buffer.resize_(gradOutput.size(1))
            torch.mv(self.buffer, gradOutput.t(), self.addBuffer).mul_(scale)
            self.gradBias.index_add_(
                1, self.partition, self.buffer.view(1, self.buffer.nelement())
            )