module 'torch' has no attribute 'sparse_csr' ,为什么在torch在1.8.1的版本中无法使用
时间: 2023-09-21 10:02:36 浏览: 651
导入FashionMNIST数据集时报错module ‘torchvision.datasets’ has no attribute ‘FashionMNIS’
在 PyTorch 1.8.1 版本中,`torch.sparse_csr_tensor()` 取代了之前的 `torch.sparse_csr()`。因此,如果你想使用稀疏矩阵 `CSR` 格式,可以使用如下的代码:
```
import torch
indices = torch.tensor([[0, 1, 1], [2, 0, 2]])
values = torch.tensor([3, 4, 5], dtype=torch.float32)
shape = torch.Size([3, 3])
sparse_tensor = torch.sparse_csr_tensor(indices, values, shape)
```
这里 `torch.sparse_csr_tensor()` 的参数和之前的 `torch.sparse_csr()` 是一样的。如果你还想使用 `torch.sparse_csr()` 可以升级 PyTorch 至最新版本,或者使用旧版本的 PyTorch。
阅读全文