class_docstr: TreeLSTM from DGL.
class_docstr_tok: ["TreeLSTM","from","DGL","."]
code: def __init__(self, hidden, in_size, out_size):
      super().__init__()
      self.emb = torch.nn.Embedding(in_size, hidden)
      self.out = torch.nn.Linear(hidden, out_size)
      self.tree_lstm = TreeLSTMCell(hidden, hidden)
      self.hidden_size = hidden
  
file_path: pytorch-struct/examples/networks/TreeLSTM.py
imports: import torch
  import torch
  import torch.nn
  import dgl
  from torch_struct import CKY
label_desc: Use image node_id 1 to create a new TreeLSTM object from inherited base classes: torch.nn with example: obj = TreeLSTM(hidden, in_size, out_size)