class_docstr: Dummy PyTorch model 
docstr: Forward method 
docstr_tok: ["Forward","method"]
code: def forward(
      self, in_dict: Dict[str, torch.Tensor]
  ) -> Dict[str, torch.Tensor]:
      """Forward method"""
      out_dict = dict()
  
      out_dict["out_key_0"] = self.cnn(in_dict["in_key_0"])
      out_dict["out_key_1"] = self.dense(in_dict["in_key_1"])
  
      return out_dict
  
file_path: maze/maze/test/perception/blocks/test_torch_model_block.py
imports: from typing import Dict
  import torch
  import torch.nn
  from maze.perception.blocks.general.torch_model_block import TorchModelBlock
  from maze.test.perception.perception_test_utils import build_multi_input_dict
label_desc: Use image node_id 2 for calling the CustomPytorchModel obj's underlying member method code with example usage: obj.forward(in_dict) and returns: out_dict