code: def __init__(self, input_size, hidden_size, num_classes):
      super(Attention, self).__init__()
      self.attention_cell = AttentionCell(
          input_size, hidden_size, num_classes
      )
      self.hidden_size = hidden_size
      self.num_classes = num_classes
      self.generator = nn.Linear(hidden_size, num_classes)
  
file_path: EasyOCR/trainer/modules/prediction.py
imports: import torch
  import torch.nn
  import torch.nn.functional
label_desc: Use image node_id 1 to create a new Attention object from inherited base classes: nn with example: obj = Attention(input_size, hidden_size, num_classes)