code: def update_r(self, node, zm=None):
      dst_x = node.data["dst_x"]
      m = node.data["m"] if zm is None else zm["m"]
      r_1 = self.W_r(dst_x)
      r_2 = self.U_r(m)
      r = torch.sigmoid(r_1 + r_2)
      return {"r": r, "rm": r * m}
  
file_path: dgl/examples/pytorch/jtnn/jtnn/nnutils.py
imports: import os
  import dgl
  import torch
  import torch.nn
label_desc: Use image node_id 3 for calling the GRUUpdate obj's underlying member method code with example usage: obj.update_r(node, zm) and returns: dict