Blockquote


def forward(self, x):
  y = self.NN1(x)
  print(x.size(), y.size()) # &lt;- this line
  x.size = batch_size,number_features
  y.size = batch_size,1 
  
  if y&gt;0:
     return self.NN2(x)
 else:
     return self.NN3(x)