def test_batch_eval_neg_styblinski_tang(self, cuda=False):
device = torch.device("cuda") if cuda else torch.device("cpu")
for dtype in (torch.float, torch.double):
X = torch.zeros(2, 3, device=device, dtype=dtype)
res = neg_styblinski_tang(X)
self.assertEqual(res.dtype, dtype)
self.assertEqual(res.device.type, device.type)
self.assertEqual(res.shape, torch.Size([2]))
self.assertTrue(torch.all(res == 0))
def test_batch_eval_neg_styblinski_tang(self, cuda=False):
device = torch.device("cuda") if cuda else torch.device("cpu")
for dtype in (torch.float, torch.double):
X = torch.zeros(2, 3, device=device, dtype=dtype)
res = neg_styblinski_tang(X)
self.assertEqual(res.dtype, dtype)
self.assertEqual(res.device.type, device.type)
self.assertEqual(res.shape, torch.Size([2]))
self.assertTrue(torch.all(res == 0))
def test_neg_styblinski_tang_global_maximum(self, cuda=False):
device = torch.device("cuda") if cuda else torch.device("cpu")
for dtype in (torch.float, torch.double):
X = torch.full(
(3,), GLOBAL_MAXIMIZER, device=device, dtype=dtype, requires_grad=True
)
res = neg_styblinski_tang(X)
res.backward()
self.assertAlmostEqual(res.item(), 3 * GLOBAL_MAXIMUM, places=4)
self.assertLess(X.grad.abs().max().item(), 1e-5)
def test_neg_styblinski_tang_global_maximum(self, cuda=False):
device = torch.device("cuda") if cuda else torch.device("cpu")
for dtype in (torch.float, torch.double):
X = torch.full((3, ),
GLOBAL_MAXIMIZER,
device=device,
dtype=dtype,
requires_grad=True)
res = neg_styblinski_tang(X)
res.backward()
self.assertAlmostEqual(res.item(), 3 * GLOBAL_MAXIMUM, places=4)
self.assertLess(X.grad.abs().max().item(), 1e-5)