def test_dirichlet_bvp_spherical(): r0, r1 = x0, x1 r2 = (r0 + r1) / 2 no_condition = NoCondition() # B.C. for the interior boundary (r_min) net_f = FCNN(2, 1) f = lambda th, ph: no_condition.enforce(net_f, th, ph) # B.C. for the exterior boundary (r_max) net_g = FCNN(2, 1) g = lambda th, ph: no_condition.enforce(net_g, th, ph) condition = DirichletBVPSpherical(r_0=r0, f=f, r_1=r1, g=g) net = FCNN(3, 1) theta = torch.rand(N_SAMPLES, 1) * np.pi phi = torch.rand(N_SAMPLES, 1) * 2 * np.pi r = r0 * ones assert all_close(condition.enforce(net, r, theta, phi), f(theta, phi)), "inner Dirichlet BC not satisfied" r = r1 * ones assert all_close(condition.enforce(net, r, theta, phi), g(theta, phi)), "inner Dirichlet BC not satisfied" condition = DirichletBVPSpherical(r_0=r2, f=f) r = r2 * ones assert all_close(condition.enforce(net, r, theta, phi), f(theta, phi)), "single ended BC not satisfied"
def test_inf_dirichlet_bvp_spherical(): r0 = random.random() r1 = 1e15 no_condition = NoCondition() net_f, net_g = FCNN(2, 1), FCNN(2, 1) # B.C. for the interior boundary (r=r_min) f = lambda th, ph: no_condition.enforce(net_f, th, ph) # B.C. for the exterior boundary (r=infinity) g = lambda th, ph: no_condition.enforce(net_g, th, ph) net = FCNN(3, 1) condition = InfDirichletBVPSpherical(r_0=r0, f=f, g=g, order=1) theta = torch.rand(10, 1) * np.pi phi = torch.rand(10, 1) * (2 * np.pi) r = r0 * ones assert all_close(condition.enforce(net, r, theta, phi), f(theta, phi)), "inner DirichletBC not satisfied" r = r1 * ones assert all_close(condition.enforce(net, r, theta, phi), g(theta, phi)), "Infinity DirichletBC not satisfied"
def test_no_condition(): N_INPUTS = 5 N_OUTPUTS = 5 for n_in, n_out in zip(range(1, N_INPUTS), range(1, N_OUTPUTS)): xs = [torch.rand(N_SAMPLES, 1, requires_grad=True) for _ in range(n_in)] net = FCNN(n_in, n_out) cond = NoCondition() y_cond = cond.enforce(net, *xs) y_raw = net(torch.cat(xs, dim=1)) assert (y_cond == y_raw).all()
def U(x): cond = NoCondition() nets = [FCNN(3, 1) for _ in range(3)] return tuple(cond.enforce(net, *x) for net in nets)
def u(x): cond = NoCondition() net = FCNN(3, 1) return cond.enforce(net, *x)
def scalar_field(x): cond = NoCondition() return cond.enforce(FCNN(3, 1), *x)
def vector_field(x): cond = NoCondition() return tuple(cond.enforce(FCNN(3, 1), *x) for _ in range(3))