code: def __init__(self, *args, scaling_correction=False, elemwise=False):
      super().__init__(*args)
      self.scaling_correction = 1.732 if scaling_correction else 1.0
      self.elemwise = elemwise
      self.Wx = nn.Linear(self.d_input, self.d_input)
      if self.elemwise:
          self.Wy = nn.Parameter(torch.randn(self.d_input))
      else:
          self.Wy = nn.Linear(self.d_input, self.d_input)
  
imports: import torch
  from torch import nn
label_desc: Use image node_id 1 to create a new Highway object from inherited base classes: Residual with example: obj = Highway()