code: def foo(self, x):
      return x.cos()
  
file_path: pytorch/torch/_export/db/examples/cond_branch_class_method.py
imports: import torch
  from torch._export.db.case import export_case
  from functorch.experimental.control_flow import cond
label_desc: Use image node_id 1 for calling the MySubModule obj's underlying member method code with example usage: obj.foo(x) and returns: x