cupyx.scipy.linalg.block_diag#

cupyx.scipy.linalg.block_diag(*arrs)[source]#

Create a block diagonal matrix from provided arrays.

Given the inputs A, B, and C, the output will have these arrays arranged on the diagonal:

[A, 0, 0]
[0, B, 0]
[0, 0, C]

Parameters:

A (cupy.ndarray) – Input arrays. A 1-D array of length n is treated as a 2-D array with shape (1,n).
B (cupy.ndarray) – Input arrays. A 1-D array of length n is treated as a 2-D array with shape (1,n).
C (cupy.ndarray) – Input arrays. A 1-D array of length n is treated as a 2-D array with shape (1,n).
... (cupy.ndarray) – Input arrays. A 1-D array of length n is treated as a 2-D array with shape (1,n).

Returns:

Array with A, B, C, … on the diagonal. Output has the same dtype as A.

Return type:

(cupy.ndarray)