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