Skip to content

Commit 18336f8

Browse files
authored
docs: remove unnecessary begin...end markers (#1567)
1 parent 48d8b9c commit 18336f8

File tree

1 file changed

+23
-25
lines changed

1 file changed

+23
-25
lines changed

examples/PINN2DPDE/main.jl

Lines changed: 23 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -116,38 +116,36 @@ analytical_solution(xyt) = analytical_solution(xyt[1, :], xyt[2, :], xyt[3, :])
116116
nothing #hide
117117
#-
118118

119-
begin
120-
grid_len = 16
119+
grid_len = 16
121120

122-
grid = range(0.0f0, 2.0f0; length=grid_len)
123-
xyt = stack([[elem...] for elem in vec(collect(Iterators.product(grid, grid, grid)))])
121+
grid = range(0.0f0, 2.0f0; length=grid_len)
122+
xyt = stack([[elem...] for elem in vec(collect(Iterators.product(grid, grid, grid)))])
124123

125-
target_data = reshape(analytical_solution(xyt), 1, :)
124+
target_data = reshape(analytical_solution(xyt), 1, :)
126125

127-
bc_len = 512
126+
bc_len = 512
128127

129-
x = collect(range(0.0f0, 2.0f0; length=bc_len))
130-
y = collect(range(0.0f0, 2.0f0; length=bc_len))
131-
t = collect(range(0.0f0, 2.0f0; length=bc_len))
128+
x = collect(range(0.0f0, 2.0f0; length=bc_len))
129+
y = collect(range(0.0f0, 2.0f0; length=bc_len))
130+
t = collect(range(0.0f0, 2.0f0; length=bc_len))
132131

133-
xyt_bc = hcat(
134-
stack((x, y, zeros(Float32, bc_len)); dims=1),
135-
stack((zeros(Float32, bc_len), y, t); dims=1),
136-
stack((ones(Float32, bc_len) .* 2, y, t); dims=1),
137-
stack((x, zeros(Float32, bc_len), t); dims=1),
138-
stack((x, ones(Float32, bc_len) .* 2, t); dims=1),
139-
)
140-
target_bc = reshape(analytical_solution(xyt_bc), 1, :)
132+
xyt_bc = hcat(
133+
stack((x, y, zeros(Float32, bc_len)); dims=1),
134+
stack((zeros(Float32, bc_len), y, t); dims=1),
135+
stack((ones(Float32, bc_len) .* 2, y, t); dims=1),
136+
stack((x, zeros(Float32, bc_len), t); dims=1),
137+
stack((x, ones(Float32, bc_len) .* 2, t); dims=1),
138+
)
139+
target_bc = reshape(analytical_solution(xyt_bc), 1, :)
141140

142-
min_target_bc, max_target_bc = extrema(target_bc)
143-
min_data, max_data = extrema(target_data)
144-
min_pde_val, max_pde_val = min(min_data, min_target_bc), max(max_data, max_target_bc)
141+
min_target_bc, max_target_bc = extrema(target_bc)
142+
min_data, max_data = extrema(target_data)
143+
min_pde_val, max_pde_val = min(min_data, min_target_bc), max(max_data, max_target_bc)
145144

146-
xyt = (xyt .- minimum(xyt)) ./ (maximum(xyt) .- minimum(xyt))
147-
xyt_bc = (xyt_bc .- minimum(xyt_bc)) ./ (maximum(xyt_bc) .- minimum(xyt_bc))
148-
target_bc = (target_bc .- min_pde_val) ./ (max_pde_val - min_pde_val)
149-
target_data = (target_data .- min_pde_val) ./ (max_pde_val - min_pde_val)
150-
end
145+
xyt = (xyt .- minimum(xyt)) ./ (maximum(xyt) .- minimum(xyt))
146+
xyt_bc = (xyt_bc .- minimum(xyt_bc)) ./ (maximum(xyt_bc) .- minimum(xyt_bc))
147+
target_bc = (target_bc .- min_pde_val) ./ (max_pde_val - min_pde_val)
148+
target_data = (target_data .- min_pde_val) ./ (max_pde_val - min_pde_val)
151149
nothing #hide
152150

153151
# ## Training

0 commit comments

Comments
 (0)