class_docstr: Simple Convolutional Neural Network
docstr: predict digit for input
docstr_tok: ["predict","digit","for","input"]
code: def predict(self, inp):
      """predict digit for input"""
      self.eval()
      with torch.no_grad():
          raw_output = self(inp)
          _, pred = torch.max(raw_output, 1)
          return pred
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 3 for calling the SimpleConvNet obj's underlying member method code with example usage: obj.predict(inp) and returns: pred