cupyx.scipy.sparse.diags#

cupyx.scipy.sparse.diags(diagonals, offsets=0, shape=None, format=None, dtype=None)[source]#

从对角线构造稀疏矩阵。

参数:

  • diagonals (序列，元素为 array_like) – 包含矩阵对角线（与 offsets 对应）的数组序列。

  • offsets (序列，元素为 int 或单个 int) –

    要设置的对角线

    • k = 0 主对角线 (默认)

    • k > 0 第 k 条上对角线

    • k < 0 第 k 条下对角线

  • shape (int 元组) – 结果的形状。如果省略，则返回一个足够大的方阵以包含对角线。

  • format ({"dia", "csr", "csc", "lil", ...}) – 结果的矩阵格式。默认情况下 (format=None) 返回适当的稀疏矩阵格式。此选择可能会更改。

  • dtype (dtype) – 矩阵的数据类型。

返回值:

生成的矩阵。

返回类型:

cupyx.scipy.sparse.spmatrix

注意

此函数在处理非对角线的方式上与 spdiags 不同。

diags 的结果是以下稀疏等价物：

cupy.diag(diagonals[0], offsets[0])
+ ...
+ cupy.diag(diagonals[k], offsets[k])

不允许重复的对角线偏移。