cupyx.scipy.sparse.dia_matrix#
- class cupyx.scipy.sparse.dia_matrix(arg1, shape=None, dtype=None, copy=False)[source]#
对角线存储的稀疏矩阵。
目前它只有以下一种初始化格式
dia_matrix((data, offsets))
- 参数:
arg1 – 初始化器的参数。
shape (tuple) – 矩阵的形状。其长度必须为二。
dtype – 数据类型。它必须是
numpy.dtype
的参数。copy (bool) – 如果为
True
,则始终使用给定数组的副本。
方法
- conj(copy=True)[source]#
逐元素复数共轭。
如果矩阵为非复数数据类型且 copy 为 False,则此方法不执行任何操作且不复制数据。
- 参数:
copy (bool) – 如果为 True,则保证结果与自身不共享数据。
- 返回:
逐元素复数共轭。
- 返回类型:
- conjugate(copy=True)[source]#
逐元素复数共轭。
如果矩阵为非复数数据类型且 copy 为 False,则此方法不执行任何操作且不复制数据。
- 参数:
copy (bool) – 如果为 True,则保证结果与自身不共享数据。
- 返回:
逐元素复数共轭。
- 返回类型:
- diagonal(k=0)[source]#
返回矩阵的第 k 个对角线。
- 参数:
k (int, optional) – 要获取的对角线,对应于元素
a[i – 0(主对角线)。
默认 (i+k].) – 0(主对角线)。
- 返回:
第 k 个对角线。
- 返回类型:
- get(stream=None)[source]#
在主机内存上返回数组的副本。
- 参数:
stream (cupy.cuda.Stream) – CUDA 流对象。如果给定,复制将异步运行。否则,复制是同步的。
- 返回:
主机内存上的数组副本。
- 返回类型:
- mean(axis=None, dtype=None, out=None)[source]#
计算指定轴上的算术平均值。
- 参数:
axis (int or
None
) – 计算平均值的轴。如果为None
,则计算所有元素的平均值。从{None, 0, 1, -2, -1}
中选择。- 返回:
求和后的数组。
- 返回类型:
参见
scipy.sparse.spmatrix.mean()
- reshape(*shape, order=‘C’)[source]#
在不改变稀疏矩阵数据的情况下赋予其新形状。
- 参数:
shape (tuple) – 新形状应与原形状兼容。
order – {‘C’, ‘F’} (可选) 使用此索引顺序读取元素。‘C’ 表示使用 C-like 索引顺序读写元素。‘F’ 表示使用 Fortran-like 索引顺序读写元素。默认值: C。
- 返回:
稀疏矩阵
- 返回类型:
- setdiag(values, k=0)[source]#
设置数组的对角线或非对角线元素。
- 参数:
values (cupy.ndarray) – 对角线元素的新值。Values 可以有任何长度。如果对角线比 values 长,则剩余的对角线条目将不会被设置。如果 values 比对角线长,则剩余的 values 被忽略。如果给定标量值,则整个对角线都将被设置为该值。
k (int, optional) – 要设置的对角线,对应于元素 a[i, i+k]。默认值: 0(主对角线)。
- sum(axis=None, dtype=None, out=None)[source]#
对给定轴上的矩阵元素求和。
- 参数:
axis (int or
None
) – 计算总和的轴。如果为None
,则计算所有元素的总和。从{None, 0, 1, -2, -1}
中选择。dtype – 返回矩阵的类型。如果未指定,则使用数组的类型。
out (cupy.ndarray) – 输出矩阵。
- 返回:
求和后的数组。
- 返回类型:
参见
scipy.sparse.spmatrix.sum()
- tocsc(copy=False)[source]#
将矩阵转换为压缩稀疏列格式。
- 参数:
copy (bool) – 如果为
False
,则尽可能共享数据数组。实际上此选项被忽略,因为在 dia 到 csc 的转换中矩阵中的所有数组无法共享。- 返回:
转换后的矩阵。
- 返回类型:
- tocsr(copy=False)[source]#
将矩阵转换为压缩稀疏行格式。
- 参数:
copy (bool) – 如果为
False
,则尽可能共享数据数组。实际上此选项被忽略,因为在 dia 到 csr 的转换中矩阵中的所有数组无法共享。- 返回:
转换后的矩阵。
- 返回类型:
属性
- H#
- T#
- device#
此数组所在的 CUDA 设备。
- dtype#
矩阵的数据类型。
- format = 'dia'#
- ndim#
- nnz#
- shape#
- size#