@@ -20,13 +20,14 @@ def test_kernel(self, cuda=False):
2020 b = torch .tensor ([[[1 , 3 ], [0 , 4 ]]], dtype = torch .float )
2121
2222 actual = torch .tensor (
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- ]
23+ [
24+ [0.3056225 , - 0.0000000 , 0.5822443 , 0.0188260 , - 0.0209871 , 0.0419742 ],
25+ [0.0000000 , 0.5822443 , 0.0000000 , 0.0209871 , - 0.0056045 , 0.0531832 ],
26+ [- 0.5822443 , 0.0000000 , - 0.8515886 , - 0.0419742 , 0.0531832 , - 0.0853792 ],
27+ [0.1304891 , - 0.2014212 , - 0.2014212 , 0.0336440 , - 0.0815567 , - 0.0000000 ],
28+ [0.2014212 , - 0.1754366 , - 0.3768578 , 0.0815567 , - 0.1870145 , - 0.0000000 ],
29+ [0.2014212 , - 0.3768578 , - 0.1754366 , 0.0000000 , - 0.0000000 , 0.0407784 ],
30+ ]
3031 )
3132
3233 kernel = Matern52KernelGrad ()
0 commit comments