code: def test_LSTMNet():
      sample_input = torch.tensor([[1, 2, 8, 3, 2], [2, 4, 99, 1, 7]])
      sample_lengths = torch.tensor([5, 5])
  
      # Simple LSTM
      model = LSTMNet(num_classes=2, dropout_prob=0.5)
      _ = model(sample_input, sample_lengths)
  
      # Bidirectional LSTM
      model = LSTMNet(
          num_classes=2, dropout_prob=0.5, bidirectional=True
      )
      _ = model(sample_input, sample_lengths)
  
file_path: KD_Lib/tests/test_models.py
imports: import torch
  from KD_Lib.models import LeNet, LSTMNet, ModLeNet, NetworkInNetwork, ResNet18, ResNet34, ResNet50, ResNet101, ResNet152, Shallow
label_desc: Use image node_id 7 for calling a global function with example usage: test_LSTMNet() without return types