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