LeastSquare.py 文件源码

def forward(self, input):
        mask_le_mone = input.le(-1).type_as(input)
        self.mask_ge_one = input.ge(1).type_as(input)
        index_gt_mone = input.gt(-1).type_as(input)
        index_lt_one = input.lt(1).type_as(input)
        self.mask_mone_one = index_lt_one * index_gt_mone
        mone = input.new().resize_as_(input).fill_(-1)
        mone= mone * mask_le_mone
        between_one = torch.pow(1 + input, 2) -1
        between_one = between_one * self.mask_mone_one
        ge_one = input * 4 - 1
        ge_one = ge_one * self.mask_ge_one
        between_one = mone + between_one
        ge_one = between_one + ge_one
        self.input = input
        return ge_one