detray-sympy/detray_sympy/checks.py

0001 def has_same_known_substructure(A, B):
0002     if A.shape != B.shape:
0003         return False
0004
0005     for i in range(A.shape[0]):
0006         for j in range(A.shape[1]):
0007             if A[i, j] == 0 and B[i, j] != 0:
0008                 return False
0009             if B[i, j] == 0 and A[i, j] != 0:
0010                 return False
0011             if A[i, j] == 1 and B[i, j] != 1:
0012                 return False
0013             if B[i, j] == 1 and A[i, j] != 1:
0014                 return False
0015
0016     return True