torch.sparse_coo_tensor
时间: 2023-04-15 07:03:08 浏览: 172
torch_sparse-0.6.10-cp39-cp39-linux_x86_64whl.zip
torch.sparse_coo_tensor是PyTorch中的一个稀疏张量类型,它是一种基于COO(Coordinate)格式的稀疏矩阵表示方法。COO格式是一种基于坐标的稀疏矩阵表示方法,它将矩阵中非零元素的坐标和值分别存储在三个数组中,分别为行坐标、列坐标和值。在PyTorch中,torch.sparse_coo_tensor可以用来表示稀疏矩阵,它支持基本的张量操作,如加、减、乘、转置等。
阅读全文