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