init(featureChannelCount:beta:gamma:varianceEpsilon:momentum:)
Creates an instance normalization layer with the number of feature channels, beta and gamma tensors, variance epsilon, and momentum you specify.
Declaration
convenience init?(featureChannelCount: Int, beta: MLCTensor?, gamma: MLCTensor?, varianceEpsilon: Float, momentum: Float)Parameters
- featureChannelCount:
The number of feature channels.
- beta:
The beta tensor.
- gamma:
The gamma tensor.
- varianceEpsilon:
The variance epsilon you use for numerical stability.
- momentum:
The momentum value for the running mean and variance computation.