GroupNorm

public final class GroupNorm : Model

Group normalization model.


                    
                    
                    init(axis:groups:epsilon:reduce:name:)

Swift

public init(axis: Int, groups: Int, epsilon: Float, reduce: [Int], name: String = "")


                    
                    
                    callAsFunction(_:streamContext:)

Swift

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