class_docstr: input 299*299*3
  output 35*35*384
class_docstr_tok: ["input","299","*","299","*","3","output","35","*","35","*","384"]
code: def forward(self, x):
      x = self.model_a(x)
      x_0 = self.branch_a0(x)
      x_1 = self.branch_a1(x)
      x = torch.cat((x_0, x_1), 1)
      x_0 = self.branch_b0(x)
      x_1 = self.branch_b1(x)
      x = torch.cat((x_0, x_1), 1)
      x_0 = self.branch_c0(x)
      x_1 = self.branch_c1(x)
      x = torch.cat((x_0, x_1), 1)
      return x
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 2 for calling the StemBlock obj's underlying member method code with example usage: obj.forward(x) and returns: x