apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-07-14 16:32:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

minor adjustment to fast_array_equal

Browse source
This commit is contained in:
Max Zwiessele 2013-07-29 15:28:44 +01:00
parent 30742ddb85
commit 92c6e652d9
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
GPy/util/misc.py
View file

@ -108,12 +108,12 @@ def fast_array_equal(A, B):
elif ((A == None) and (B != None)) or ((A != None) and (B == None)): elif ((A == None) and (B != None)) or ((A != None) and (B == None)):
return False return False
elif A.shape == B.shape: elif A.shape == B.shape:
if len(A.shape) == 2: if A.ndim == 2:
N, D = A.shape N, D = A.shape
value = weave.inline(code2, support_code=support_code, libraries=['gomp'], value = weave.inline(code2, support_code=support_code, libraries=['gomp'],
arg_names=['A', 'B', 'N', 'D'], arg_names=['A', 'B', 'N', 'D'],
type_converters=weave.converters.blitz,**weave_options) type_converters=weave.converters.blitz,**weave_options)
elif len(A.shape) == 3: elif A.ndim == 3:
N, D, Q = A.shape N, D, Q = A.shape
value = weave.inline(code3, support_code=support_code, libraries=['gomp'], value = weave.inline(code3, support_code=support_code, libraries=['gomp'],
arg_names=['A', 'B', 'N', 'D', 'Q'], arg_names=['A', 'B', 'N', 'D', 'Q'],