diff options
Diffstat (limited to 'Demos/benchmarks/spectralnorm.py')
| -rw-r--r-- | Demos/benchmarks/spectralnorm.py | 22 |
1 files changed, 11 insertions, 11 deletions
diff --git a/Demos/benchmarks/spectralnorm.py b/Demos/benchmarks/spectralnorm.py index 7b56b05f6..f476c6bcd 100644 --- a/Demos/benchmarks/spectralnorm.py +++ b/Demos/benchmarks/spectralnorm.py @@ -11,28 +11,28 @@ from time import time import util import optparse -def eval_A (i, j): +def eval_A(i, j): return 1.0 / ((i + j) * (i + j + 1) / 2 + i + 1) -def eval_A_times_u (u): +def eval_A_times_u(u): return [ part_A_times_u(i,u) for i in range(len(u)) ] -def eval_At_times_u (u): +def eval_At_times_u(u): return [ part_At_times_u(i,u) for i in range(len(u)) ] -def eval_AtA_times_u (u): - return eval_At_times_u (eval_A_times_u (u)) +def eval_AtA_times_u(u): + return eval_At_times_u(eval_A_times_u(u)) def part_A_times_u(i, u): partial_sum = 0 for j, u_j in enumerate(u): - partial_sum += eval_A (i, j) * u_j + partial_sum += eval_A(i, j) * u_j return partial_sum def part_At_times_u(i, u): partial_sum = 0 for j, u_j in enumerate(u): - partial_sum += eval_A (j, i) * u_j + partial_sum += eval_A(j, i) * u_j return partial_sum DEFAULT_N = 130 @@ -43,13 +43,13 @@ def main(n): t0 = time() u = [1] * DEFAULT_N - for dummy in range (10): - v = eval_AtA_times_u (u) - u = eval_AtA_times_u (v) + for dummy in range(10): + v = eval_AtA_times_u(u) + u = eval_AtA_times_u(v) vBv = vv = 0 - for ue, ve in zip (u, v): + for ue, ve in zip(u, v): vBv += ue * ve vv += ve * ve tk = time() |