42e617d6b759af70101cd41f5b28f5d242be7c84,beginner_source/Intro_to_TorchScript_tutorial.py,MyCell,forward,#MyCell#Any#Any#,271

Before Change

        self.linear = torch.nn.Linear(4, 4)

    def forward(self, x, h):
        new_h = torch.tanh(self.dg(self.linear(x)) + h)
        return new_h, new_h

my_cell = MyCell(MyDecisionGate())

After Change

        self.linear = torch.nn.Linear(4, 4)

    def forward(self, x, h):
        new_h = torch.tanh(self.linear(x) + h)
        return new_h, new_h

my_cell = MyCell()

Instances