LayerNorm

public final class LayerNorm : Model

Layer normalization model.


                    
                    
                    init(epsilon:axis:elementwiseAffine:trainable:name:)

Swift

public init(
  epsilon: Float, axis: [Int], elementwiseAffine: Bool = true, trainable: Bool? = nil,
  name: String = ""
)


                    
                    
                    callAsFunction(_:streamContext:)

Swift

public func callAsFunction<T: DynamicGraph.TensorGroup>(
  _ input: T, streamContext: StreamContext? = nil
) -> T