class_docstr: Tests based on a more non-standard model
  
  Specifically, add modules with non-standard names and non-standard outputs
  like tuples or dicts.
  
  This test class does not re-iterate all the tests performed on the standard
  model but focuses on the parts that change, e.g. how the outputs are
  recorded.
class_docstr_tok: ["Tests","based","on","a","more","non-standard","model","Specifically",",","add","modules","with","non-standard","names","and","non-standard","outputs","like","tuples","or","dicts",".","This","test","class","does","not","re-iterate","all","the","tests","performed","on","the","standard","model","but","focuses","on","the","parts","that","change",",","e.g",".","how","the","outputs","are","recorded","."]
docstr: Module that combines module0 and module1
docstr_tok: ["Module","that","combines","module0","and","module1"]
code: def module2_cls(self, module0_cls, module1_cls):
      """Module that combines module0 and module1"""
  
      class MyModule(nn.Module):
          """Module that returns a dict"""
  
          def __init__(self):
              super().__init__()
              self.module0 = module0_cls()
              self.module1 = module1_cls()
  
          def forward(self, X):
              _, X1 = self.module0(X)
              output = self.module1(X1)
              return output["softmax"]
  
      return MyModule
  
imports: import itertools
  from unittest import mock
  import numpy
  import pytest
  import torch
  from torch import nn
label_desc: Use image node_id 3 for calling the TestSkorchDoctorComplexArchitecture obj's underlying member method code with example usage: obj.module2_cls(module0_cls, module1_cls) and returns: MyModule, output