Contents

init(featureChannelCount:mean:variance:beta:gamma:varianceEpsilon:)

Creates a batch normalization layer with the number of feature channels, tensors, and variance epsilon you specify.

Declaration

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

Parameters

  • featureChannelCount:

    The number of feature channels.

  • mean:

    The mean tensor.

  • variance:

    The variance tensor.

  • beta:

    The beta tensor.

  • gamma:

    The gamma tensor.

  • varianceEpsilon:

    The variance epsilon you use for numerical stability.

See Also

Creating Batch Normalization Layers