LayerNorm
public final class LayerNorm : Model
Layer normalization model.
-
Declaration
Swift
public init( epsilon: Float, axis: [Int], elementwiseAffine: Bool = true, trainable: Bool? = nil, name: String = "" )
-
Declaration
Swift
public func callAsFunction<T: DynamicGraph.TensorGroup>( _ input: T, streamContext: StreamContext? = nil ) -> T