Function SyncBatchNormalization

layer/SyncBatchNormalization.go:30–49

()

}

func SyncBatchNormalization() *LSyncBatchNormalization {
	return &LSyncBatchNormalization{
		axis:                      -1,
		betaConstraint:            &constraint.NilConstraint{},
		betaInitializer:           initializer.Zeros(),
		betaRegularizer:           &regularizer.NilRegularizer{},
		center:                    true,
		dtype:                     Float32,
		epsilon:                   0.001,
		gammaConstraint:           &constraint.NilConstraint{},
		gammaInitializer:          initializer.Ones(),
		gammaRegularizer:          &regularizer.NilRegularizer{},
		momentum:                  0.99,
		movingMeanInitializer:     initializer.Zeros(),
		movingVarianceInitializer: initializer.Ones(),
		name:                      UniqueName("sync_batch_normalization"),
		scale:                     true,
		trainable:                 true,
	}
}

func (l *LSyncBatchNormalization) SetAxis(axis float64) *LSyncBatchNormalization {
	l.axis = axis