@@ -116,38 +116,36 @@ analytical_solution(xyt) = analytical_solution(xyt[1, :], xyt[2, :], xyt[3, :])
116116nothing # 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)
151149nothing # hide
152150
153151# ## Training
0 commit comments