GroupNorm

Swift

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

Swift

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