Contents

init(featureChannelCount:groupCount:beta:gamma:varianceEpsilon:)

Creates a group normalization layer with the number of feature channels and groups, beta and gamma tensors, and variance epsilon you specify.

Declaration

convenience init?(featureChannelCount: Int, groupCount: Int, beta: MLCTensor?, gamma: MLCTensor?, varianceEpsilon: Float)

Parameters

  • featureChannelCount:

    The number of feature channels.

  • groupCount:

    The number of groups into which you separate the channels.

  • beta:

    The beta tensor.

  • gamma:

    The gamma tensor.

  • varianceEpsilon:

    The variance epsilon you use for numerical stability.