This commit is contained in:
Jan Petykiewicz 2024-03-18 11:00:50 -07:00
parent 711ce119d0
commit 9ef62b92d5

View File

@ -274,7 +274,7 @@ def connect_s(
Bnew = numpy.delete(Bnew, (l,), 2) Bnew = numpy.delete(Bnew, (l,), 2)
dtype = (A[0, 0] * B[0, 0]).dtype dtype = (A[0, 0] * B[0, 0]).dtype
C = numpy.zeros(tuple(A.shape[:-2]) + (nn, nn), dtype=dtype) C = numpy.zeros(tuple(A.shape[:-2]) + (nC, nC), dtype=dtype)
C[..., :nA - 1, :nA - 1] = Anew C[..., :nA - 1, :nA - 1] = Anew
C[..., nA - 1:, nA - 1:] = Bnew C[..., nA - 1:, nA - 1:] = Bnew
return C return C