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 without learnable params that returns a dict
docstr_tok: ["Module","without","learnable","params","that","returns","a","dict"]
code: def module1_cls(self):
      """Module without learnable params that returns a dict"""
  
      class MyModule(nn.Module):
          """Module that returns a dict"""
  
          def __init__(self):
              super().__init__()
              self.softmax = nn.Softmax(dim=-1)
  
          def forward(self, X):
              softmax = self.softmax(X)
              return {"logits": X, "softmax": 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 2 for calling the TestSkorchDoctorComplexArchitecture obj's underlying member method code with example usage: obj.module1_cls() and returns: MyModule, dict