Skip to content

Commit 6f80199

Browse files
committed
Update transformer_poisson.py
1 parent 96c6f88 commit 6f80199

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

doc/src/week10/programs/transformer_poisson.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,7 @@ def make_sample(n, k_max=K_MAX, rng=None):
148148
rng = np.random
149149
x = interior_grid(n)
150150
ks = np.arange(1, k_max + 1)
151-
amps = rng.randn(k_max)
151+
amps = rng.standard_normal(k_max)
152152
# Normalise so the solution has unit RMS for numerical stability
153153
f_vals = np.sum(amps[:, None] * np.sin(np.pi * ks[:, None] * x[None, :]),
154154
axis=0)

0 commit comments

Comments
 (0)