scipy的稀疏矩阵转换成torch的sparse tensor

作者: heishanlaoniu | 来源:发表于2021-01-10 21:13 被阅读0次

scipy的稀疏矩阵转换成torch的sparse tensor
稀疏矩阵 scipy.sparse
sparse scipy 从稀疏矩阵中选取某些指定列构成新的稀疏
csc_matrix
pytorch 常用数据类型转换
Pytorch自学笔记
sparse
利用CRNN来识别图片中的文字（二）tensorflow中ctc
PyTorch 快速上手
torch中的乘法函数


def scipy_sparse_mat_to_torch_sparse_tensor(sparse_mx):
    """
    将scipy的sparse matrix转换成torch的sparse tensor.
    """
    sparse_mx = sparse_mx.tocoo().astype(np.float32)
    indices = torch.from_numpy(
        np.vstack((sparse_mx.row, sparse_mx.col)).astype(np.int64))
    values = torch.from_numpy(sparse_mx.data)
    shape = torch.Size(sparse_mx.shape)
    return torch.sparse.FloatTensor(indices, values, shape)

scipy的稀疏矩阵转换成torch的sparse tensor
稀疏矩阵 scipy.sparse
CSR： row offset : 最后一位是非0元素个数。其他从前至后依次为matrix中每行第一个非0元素的在...
sparse scipy 从稀疏矩阵中选取某些指定列构成新的稀疏
sparse scipy 从稀疏矩阵中选取某些指定列构成新的稀疏矩阵主要想法：1.getcol(i)获得指定行2....
csc_matrix
许多同学可能在使用Python进行科学计算时用过稀疏矩阵的构造，而python的科学计算包scipy.sparse...
pytorch 常用数据类型转换
torch.Tensor torch.Tensor是一种包含单一数据类型元素的多维矩阵。torch.Tensor是...
Pytorch自学笔记
1.1 创建矩阵 torch.tensor(data) torch.Tensor(dim1,dim2) #也可以是...
sparse
创建稀疏矩阵语法S = sparse(A)S = sparse(m,n)S = sparse(i,j,v)S =...
利用CRNN来识别图片中的文字（二）tensorflow中ctc
1. tf.SparseTensor() 定义一个稀疏tensor。 2. tf.sparse_tensor_to...
PyTorch 快速上手
torch.Tensor数据类型 torch.Tensor是一种包含单一数据类型元素的多维矩阵。 torch.Te...
torch中的乘法函数
torch.dot() torch.mm() 只能让两个二维tensor作矩阵乘法。 torch.matmul()...