LayerNorm

public var offset: Tensor<Scalar>

public var scale: Tensor<Scalar>

@noDerivative
public let axis: Int

@noDerivative
public let epsilon: Scalar

public init(
  offset: Tensor<Scalar>,
  scale: Tensor<Scalar>,
  axis: Int,
  epsilon: Scalar
)

public init(
  featureCount: Int,
  axis: Int,
  epsilon: Scalar = 0.001
)

@differentiable
public func forward(_ input: Tensor<Scalar>) -> Tensor<Scalar>