torch_geometric.transforms.ToDense

class ToDense(num_nodes: Optional[int] = None)[source]

Bases: BaseTransform

将稀疏邻接矩阵转换为形状为[num_nodes, num_nodes, *]的密集邻接矩阵(函数名称:to_dense)。

Parameters:

num_nodes (int, optional) – 节点的数量。如果设置为 None, 节点的数量将自动推断。 (默认值: None)