(tensor, mean, variance, beta, gamma, scale)
| 59 | |
| 60 | |
| 61 | def batch_norm_slow(tensor, mean, variance, beta, gamma, scale): |
| 62 | batch_norm = (tensor - mean) * math_ops.rsqrt(variance + 0.001) |
| 63 | if scale: |
| 64 | batch_norm *= gamma |
| 65 | return batch_norm + beta |
| 66 | |
| 67 | |
| 68 | def build_graph(device, input_shape, axes, num_layers, mode, scale, train): |