auto_reg_nn.py 文件源码

def forward(self, _input):
        """
        the forward method that does the masked linear computation and returns the result
        """
        masked_weight = self.weight * torch.autograd.Variable(self.mask)
        return F.linear(_input, masked_weight, self.bias)