code: def node_update_func(self, nodes):
      x, h, m, c = (
          nodes.data["x"],
          nodes.data["rnn_h"],
          nodes.data["m"],
          nodes.data["rnn_c"],
      )
      new_h, new_c = self.lstm(torch.cat([x, m], -1), (h, c))
      return {"h": new_h, "rnn_c": new_c, "rnn_h": new_h}
  
file_path: dgl/examples/pytorch/rrn/sudoku.py
imports: import torch
  from rrn import RRN
  from torch import nn
label_desc: Use image node_id 3 for calling the SudokuNN obj's underlying member method code with example usage: obj.node_update_func(nodes) and returns: dict