Python implementation of batch normalization.
(tensor, mean, variance, beta, gamma, scale)
| 53 | # batch_norm *= gamma |
| 54 | # return batch_norm + beta |
| 55 | def batch_norm_py(tensor, mean, variance, beta, gamma, scale): |
| 56 | """Python implementation of batch normalization.""" |
| 57 | return nn_impl.batch_normalization(tensor, mean, variance, beta, gamma if |
| 58 | scale else None, 0.001) |
| 59 | |
| 60 | |
| 61 | def batch_norm_slow(tensor, mean, variance, beta, gamma, scale): |