weixin_41722144的博客

原创 【pytorch】RuntimeError: number of dimensions must be sparse_dim (14) + dense_dim (0), but got 2

假设我们要定义一个稀疏张量，其中条目 3 在位置 (0, 2)，条目 4 在位置 (1, 0)，条目 5 在位置 (1, 2)。假定未指定的元素具有相同的值，填充值，默认情况下为零。写代码时遇到如标题所示报错，pytorch=1.7.1，使用torch.sprase_coo_tensor创建稀疏矩阵。请注意，输入i不是索引元组列表。如果你想这样写你的索引，你应该在将它们传递给稀疏构造函数之前进行。查看pytorch文档找到sparse部分，故对indices进行一个转置即可。