Fixed documentation for batch_normalization (#4689)
This commit is contained in:
parent
d246219976
commit
89d1d5c449
@ -936,9 +936,9 @@ def batch_normalization(x,
|
|||||||
|
|
||||||
As described in http://arxiv.org/abs/1502.03167.
|
As described in http://arxiv.org/abs/1502.03167.
|
||||||
Normalizes a tensor by `mean` and `variance`, and applies (optionally) a
|
Normalizes a tensor by `mean` and `variance`, and applies (optionally) a
|
||||||
`scale` \\\\(\gamma\\\\) to it, as well as an `offset` \\\\(\\beta\\\\):
|
`scale` \\(\gamma\\) to it, as well as an `offset` \\(\beta\\):
|
||||||
|
|
||||||
\\\\(\\frac{\gamma(x-\mu)}{\sigma}+\\beta\\\\)
|
\\(\frac{\gamma(x-\mu)}{\sigma}+\beta\\)
|
||||||
|
|
||||||
`mean`, `variance`, `offset` and `scale` are all expected to be of one of two
|
`mean`, `variance`, `offset` and `scale` are all expected to be of one of two
|
||||||
shapes:
|
shapes:
|
||||||
@ -964,9 +964,9 @@ def batch_normalization(x,
|
|||||||
x: Input `Tensor` of arbitrary dimensionality.
|
x: Input `Tensor` of arbitrary dimensionality.
|
||||||
mean: A mean `Tensor`.
|
mean: A mean `Tensor`.
|
||||||
variance: A variance `Tensor`.
|
variance: A variance `Tensor`.
|
||||||
offset: An offset `Tensor`, often denoted \\\\(\\beta\\\\) in equations, or
|
offset: An offset `Tensor`, often denoted \\(\beta\\) in equations, or
|
||||||
None. If present, will be added to the normalized tensor.
|
None. If present, will be added to the normalized tensor.
|
||||||
scale: A scale `Tensor`, often denoted \\\\(\gamma\\\\) in equations, or
|
scale: A scale `Tensor`, often denoted \\(\gamma\\) in equations, or
|
||||||
`None`. If present, the scale is applied to the normalized tensor.
|
`None`. If present, the scale is applied to the normalized tensor.
|
||||||
variance_epsilon: A small float number to avoid dividing by 0.
|
variance_epsilon: A small float number to avoid dividing by 0.
|
||||||
name: A name for this operation (optional).
|
name: A name for this operation (optional).
|
||||||
|
Loading…
Reference in New Issue
Block a user