@@ -20,14 +20,13 @@ def test_kernel(self, cuda=False):
2020 b = torch .tensor ([[[1 , 3 ], [0 , 4 ]]], dtype = torch .float )
2121
2222 actual = torch .tensor (
23- [
24- [0.3056 , - 0.0000 , 0.5822 , 0.0188 , - 0.0210 , 0.0420 ],
25- [0.0000 , 0.5822 , 0.0000 , 0.0210 , - 0.0056 , 0.0532 ],
26- [- 0.5822 , 0.0000 , - 0.8516 , - 0.0420 , 0.0532 , - 0.0854 ],
27- [0.1305 , - 0.2014 , - 0.2014 , 0.0336 , - 0.0816 , - 0.0000 ],
28- [0.2014 , - 0.1754 , - 0.3769 , 0.0816 , - 0.1870 , - 0.0000 ],
29- [0.2014 , - 0.3769 , - 0.1754 , 0.0000 , - 0.0000 , 0.0408 ],
30- ],
23+ [[ 0.3056225 , - 0.0000000 , 0.5822443 , 0.0188260 , - 0.0209871 , 0.0419742 ],
24+ [ 0.0000000 , 0.5822443 , 0.0000000 , 0.0209871 , - 0.0056045 , 0.0531832 ],
25+ [- 0.5822443 , 0.0000000 , - 0.8515886 , - 0.0419742 , 0.0531832 , - 0.0853792 ],
26+ [ 0.1304891 , - 0.2014212 , - 0.2014212 , 0.0336440 , - 0.0815567 , - 0.0000000 ],
27+ [ 0.2014212 , - 0.1754366 , - 0.3768578 , 0.0815567 , - 0.1870145 , - 0.0000000 ],
28+ [ 0.2014212 , - 0.3768578 , - 0.1754366 , 0.0000000 , - 0.0000000 , 0.0407784 ]
29+ ]
3130 )
3231
3332 kernel = Matern52KernelGrad ()
0 commit comments