class_docstr: Base class for any torch model.
  
  :param device: Device the networks should be located on (cpu or cuda)
class_docstr_tok: ["Base","class","for","any","torch","model",".",":","param","device",":","Device","the","networks","should","be","located","on","(","cpu","or","cuda",")"]
docstr: Returns the device the networks are located on.
docstr_tok: ["Returns","the","device","the","networks","are","located","on","."]
code: def device(self) -> str:
      """Returns the device the networks are located on."""
      return self._device
  
file_path: maze/maze/core/agent/torch_model.py
imports: from abc import abstractmethod
  from typing import List, Dict
  import torch
label_desc: Use image node_id 3 for calling the TorchModel obj's underlying member method code with example usage: obj.device() and returns: self