torch_geometric.nn.models.InnerProductDecoder

class InnerProductDecoder

Bases: Module

来自“变分图自编码器”论文的内积解码器。

\[\sigma(\mathbf{Z}\mathbf{Z}^{\top})\]

其中 \(\mathbf{Z} \in \mathbb{R}^{N \times d}\) 表示由编码器产生的潜在空间。

forward(z: Tensor, edge_index: Tensor, sigmoid: bool = True) → Tensor

将潜在变量 z 解码为给定节点对 edge_index 的边缘概率。

Parameters:

• z (torch.Tensor) – The latent space \(\mathbf{Z}\).

• edge_index (torch.Tensor) – The edge indices.

• sigmoid (bool, 可选) – 如果设置为 False，则不会对输出应用逻辑 sigmoid 函数。 (默认: True)

Return type:

Tensor

forward_all(z: Tensor, sigmoid: bool = True) → Tensor

将潜在变量 z 解码为概率密集邻接矩阵。

Parameters:

• z (torch.Tensor) – The latent space \(\mathbf{Z}\).

• sigmoid (bool, 可选) – 如果设置为 False，则不会对输出应用逻辑 sigmoid 函数。 (默认: True)

Return type:

Tensor