flake8

try:

  import torch

  import torch.nn as nn

    from torch.nn.init import _calculate_fan_in_and_fan_out, _no_grad_normal_, _no_grad_uniform_

except:

  raise ImportError('These classes require Torch to be installed.')

class LayerNorm(nn.Module):

  """Apply Layer Normalization to input.