experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #27591 from BradHuang1999:doc-patch-batch-normalization

Browse Source 
PiperOrigin-RevId: 246171358
This commit is contained in:
TensorFlower Gardener 2019-05-01 12:41:53 -07:00
commit 935c86ac55
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/python/ops/nn_impl.py
View File

@ -1192,7 +1192,6 @@ def batch_normalization(x,
name=None):
r"""Batch normalization.
As described in [Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift](http://arxiv.org/abs/1502.03167).
Normalizes a tensor by `mean` and `variance`, and applies (optionally) a
`scale` \\(\gamma\\) to it, as well as an `offset` \\(\beta\\):
@ -1218,6 +1217,10 @@ def batch_normalization(x,
`tf.nn.moments(..., keep_dims=False)` during training, or running averages
thereof during inference.
See Source: [Batch Normalization: Accelerating Deep Network Training by
Reducing Internal Covariate Shift; S. Ioffe, C. Szegedy]
(http://arxiv.org/abs/1502.03167).
Args:
x: Input `Tensor` of arbitrary dimensionality.
mean: A mean `Tensor`.
@ -1255,7 +1258,9 @@ def fused_batch_norm(
name=None):
r"""Batch normalization.
As described in http://arxiv.org/abs/1502.03167.
See Source: [Batch Normalization: Accelerating Deep Network Training by
Reducing Internal Covariate Shift; S. Ioffe, C. Szegedy]
(http://arxiv.org/abs/1502.03167).
Args:
x: Input `Tensor` of 4 dimensions.