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