pytorch的不规则张量

在 PyTorch 中，不规则张量（Irregular Tensor）通常指的是稀疏张量（Sparse Tensor），即只有部分元素非零的张量。在 PyTorch 中，可以使用 `torch.sparse` 模块创建和处理稀疏张量。

创建稀疏张量的方式有多种，其中一种常用的方式是使用索引和值的列表来初始化稀疏张量。例如，下面的代码创建了一个 $3 \times 3$ 的稀疏张量，其中只有两个元素非零：

import torch

# 创建稀疏张量
indices = torch.LongTensor([[0, 1], [2, 0]])
values = torch.FloatTensor([3, 4])
sparse_tensor = torch.sparse.FloatTensor(indices.t(), values, torch.Size([3, 3]))

# 打印稀疏张量
print(sparse_tensor)

输出结果为：

tensor(indices=tensor([[0, 1],
                       [2, 0]]),
       values=tensor([3., 4.]),
       size=(3, 3), nnz=2, layout=torch.sparse_coo)

其中 `indices` 表示非零元素的坐标，`values` 表示非零元素的值，`size` 表示稀疏张量的形状，`nnz` 表示非零元素的个数，`layout` 表示稀疏张量的布局（这里使用的是 COO 布局）。

对于稀疏张量的操作，PyTorch 提供了许多和普通张量相似的函数和方法，例如 `torch.sparse.mm` 函数用于稠密张量和稀疏张量之间的矩阵乘法，`to_dense` 方法用于将稀疏张量转换为稠密张量等等。