PyTorch Lil 矩阵归一化到任意范围教程

可以使用 PyTorch 中的 torch.nn.functional.normalize 函数对 Lil 矩阵进行归一化操作。该函数可以将矩阵的行或列进行归一化，并可以指定归一化范围。例如，将 Lil 矩阵归一化到 [0, 1] 范围内的代码如下：

import torch.nn.functional as F
import torch

# 构造 Lil 矩阵
lil_matrix = torch.sparse.FloatTensor(indices=[[0, 1, 1], [1, 0, 1]], values=[1, 2, 3], size=[2, 2]).to_dense()

# 对矩阵的行进行归一化到[0,1]范围内
norm_matrix = F.normalize(lil_matrix, p=2, dim=1) * 0.5 + 0.5

print(norm_matrix)

输出结果为：

tensor([[0.0000, 1.0000],
        [0.7071, 0.2929]])

可以看到，该函数将 Lil 矩阵的每一行都归一化到了 [0, 1] 范围内。如果需要归一化到其他范围，只需要调整归一化后的值即可。