cupyx.scipy.linalg.block_diag

cupyx.scipy.linalg.block_diag(*arrs)

从提供的数组创建块对角矩阵。

给定输入 A, B, 和 C，输出将把这些数组排列在对角线上。

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

参数:

• A (cupy.ndarray) – 输入数组。长度为 n 的 1-D 数组被视为形状为 (1,n) 的 2-D 数组。

• B (cupy.ndarray) – 输入数组。长度为 n 的 1-D 数组被视为形状为 (1,n) 的 2-D 数组。

• C (cupy.ndarray) – 输入数组。长度为 n 的 1-D 数组被视为形状为 (1,n) 的 2-D 数组。

• ... (cupy.ndarray) – 输入数组。长度为 n 的 1-D 数组被视为形状为 (1,n) 的 2-D 数组。

返回:

对角线上包含 A, B, C, … 的数组。输出与 A 具有相同的数据类型 (dtype)。

返回类型:

(cupy.ndarray)

另请参阅

scipy.linalg.block_diag()