code: def __init__(self, dim):
      super(LayerTest, self).__init__()
      self.softmax = nn.Softmax(dim=dim)
  
file_path: pytorch2keras/tests/layers/activations/softmax.py
imports: import numpy
  import torch
  import torch.nn
  from torch.autograd import Variable
  from pytorch2keras.converter import pytorch_to_keras
label_desc: Use image node_id 1 to create a new LayerTest object from inherited base classes: nn with example: obj = LayerTest(dim)