Method foo

torch/_export/db/examples/cond_branch_class_method.py:8–9

(self, x)

class MySubModule(torch.nn.Module):
    def foo(self, x):
        return x.cos()

    def forward(self, x):
        return self.foo(x)