layer_norm

dragon.vm.torch.nn.functional.layer_norm(
  input,
  normalized_shape,
  weight,
  bias,
  eps=1e-05
)[source]

Apply the layer normalization to input. [Ba et.al, 2016]

Parameters:
Returns:

dragon.vm.torch.Tensor – The output tensor.