code: def __init__(self, function):
      super().__init__()
      self._func = function
  
imports: import torch.nn
label_desc: Use image node_id 1 to create a new LambdaLayer object from inherited base classes: nn with example: obj = LambdaLayer(function)