mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-04-28 22:36:24 +02:00
minor adjustment to fast_array_equal
This commit is contained in:
parent
30742ddb85
commit
92c6e652d9
1 changed files with 2 additions and 2 deletions
|
|
@ -108,12 +108,12 @@ def fast_array_equal(A, B):
|
|||
elif ((A == None) and (B != None)) or ((A != None) and (B == None)):
|
||||
return False
|
||||
elif A.shape == B.shape:
|
||||
if len(A.shape) == 2:
|
||||
if A.ndim == 2:
|
||||
N, D = A.shape
|
||||
value = weave.inline(code2, support_code=support_code, libraries=['gomp'],
|
||||
arg_names=['A', 'B', 'N', 'D'],
|
||||
type_converters=weave.converters.blitz,**weave_options)
|
||||
elif len(A.shape) == 3:
|
||||
elif A.ndim == 3:
|
||||
N, D, Q = A.shape
|
||||
value = weave.inline(code3, support_code=support_code, libraries=['gomp'],
|
||||
arg_names=['A', 'B', 'N', 'D', 'Q'],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue