batchNorm(mean:variance:epsilon:)
Adds a batch normalization operation to the current graph.
Declaration
func batchNorm(mean: some BNNSGraph.Builder.OperationParameter<T>, variance: some BNNSGraph.Builder.OperationParameter<T>, epsilon: Float = .ulpOfOne.squareRoot()) -> BNNSGraph.Builder.Tensor<T>Parameters
- mean:
The precomputed mean values.
- variance:
The precomputed variances.
- epsilon:
The epsilon value that the function uses to avoid division by zero.