def forward(self, inp): if not (type(inp) == Variable): inp = Variable(inp[0]) output = torch.mm(inp, F.softplus(self.l1)) #output = output + self.b1 return output