Docs4dev
Docs
tensorflow / 2.9 / raw_ops / fusedbatchnormv2.html

tf.raw_ops.FusedBatchNormV2

Batch normalization.

tf.raw_ops.FusedBatchNormV2(
    x,
    scale,
    offset,
    mean,
    variance,
    epsilon=0.0001,
    exponential_avg_factor=1,
    data_format='NHWC',
    is_training=True,
    name=None
)

Note that the size of 4D Tensors are defined by either "NHWC" or "NCHW". The size of 1D Tensors matches the dimension C of the 4D Tensors.

Args
x A Tensor. Must be one of the following types: half, bfloat16, float32. A 4D Tensor for input data.
scale A Tensor. Must be one of the following types: float32. A 1D Tensor for scaling factor, to scale the normalized x.
offset A Tensor. Must have the same type as scale. A 1D Tensor for offset, to shift to the normalized x.
mean A Tensor. Must have the same type as scale. A 1D Tensor for population mean. Used for inference only; must be empty for training.
variance A Tensor. Must have the same type as scale. A 1D Tensor for population variance. Used for inference only; must be empty for training.
epsilon An optional float. Defaults to 0.0001. A small float number added to the variance of x.
exponential_avg_factor An optional float. Defaults to 1.
data_format An optional string from: "NHWC", "NCHW". Defaults to "NHWC". The data format for x and y. Either "NHWC" (default) or "NCHW".
is_training An optional bool. Defaults to True. A bool value to indicate the operation is for training (default) or inference.
name A name for the operation (optional).
Returns
A tuple of Tensor objects (y, batch_mean, batch_variance, reserve_space_1, reserve_space_2).
y A Tensor. Has the same type as x.
batch_mean A Tensor. Has the same type as scale.
batch_variance A Tensor. Has the same type as scale.
reserve_space_1 A Tensor. Has the same type as scale.
reserve_space_2 A Tensor. Has the same type as scale.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/FusedBatchNormV2