Skip to content

Commit 68b4fe9

Browse files
committed
add empty line and change variable names
1 parent cead384 commit 68b4fe9

File tree

1 file changed

+12
-12
lines changed

1 file changed

+12
-12
lines changed

deepxde/data/pde_operator.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -272,25 +272,25 @@ def forward_call(trunk_input):
272272
losses = [loss_fn(bkd.zeros_like(error), error) for error in error_f] # noqa
273273

274274
# BC loss
275-
for k, bc in enumerate(self.pde.bcs):
276-
beg, end = bcs_start[k], bcs_start[k + 1]
277-
error_k = []
275+
for j, bc in enumerate(self.pde.bcs):
276+
beg, end = bcs_start[j], bcs_start[j + 1]
277+
error = []
278278
for i in range(num_func):
279-
output_i = outputs[i]
280-
if bkd.ndim(output_i) == 1: # noqa
281-
output_i = output_i[:, None]
282-
error_ki = bc.error(
279+
out = outputs[i]
280+
if bkd.ndim(out) == 1: # noqa
281+
out = out[:, None]
282+
error_i = bc.error(
283283
self.train_x[1],
284284
inputs[1],
285-
output_i,
285+
out,
286286
beg,
287287
end,
288288
aux_var=model.net.auxiliary_vars[i][:, None],
289289
)
290-
error_k.append(error_ki)
291-
error_k = bkd.stack(error_k, axis=0) # noqa
292-
loss_k = loss_fn(bkd.zeros_like(error_k), error_k) # noqa
293-
losses.append(loss_k)
290+
error.append(error_i)
291+
error = bkd.stack(error, axis=0) # noqa
292+
loss = loss_fn(bkd.zeros_like(error), error) # noqa
293+
losses.append(loss)
294294
return losses
295295

296296
def losses_train(self, targets, outputs, loss_fn, inputs, model, aux=None):

0 commit comments

Comments
 (0)