class_docstr: PyTorch reader to extract weights data.
class_docstr_tok: ["PyTorch","reader","to","extract","weights","data","."]
code: def __init__(self, config):
      self.torch_model = config["PytorchModel"]
      self.state_dict = self.torch_model.state_dict()
      self.input_shape = config["InputShape"]
  
file_path: hls4ml/hls4ml/converters/pytorch_to_hls.py
imports: import torch
  from hls4ml.model import ModelGraph
label_desc: Use image node_id 1 to create a new PyTorchModelReader object  with example: obj = PyTorchModelReader(config)