torch_geometric.nn.models.VGAE

class VGAE(encoder: Module, decoder: Optional[Module] = None)

Bases: GAE

来自“变分图自编码器”论文的变分图自编码器模型。

Parameters:

• encoder (torch.nn.Module) – 用于计算 \(\mu\) 和 \(\log\sigma^2\) 的编码器模块。

• decoder (torch.nn.Module, optional) – The decoder module. If set to None, will default to the torch_geometric.nn.models.InnerProductDecoder. (default: None)

forward(*args, **kwargs) → Tensor

Alias for encode().

Return type:

Tensor

reset_parameters()

重置模块的所有可学习参数。

kl_loss(mu: Optional[Tensor] = None, logstd: Optional[Tensor] = None) → Tensor

计算KL损失，基于传递的参数mu和logstd，或基于上次编码的潜在变量。

Parameters:

• mu (torch.Tensor, optional) – 潜在空间的 \(\mu\)。如果设置为 None，则使用最后一次计算的 \(\mu\)。 (默认值: None)

• logstd (torch.Tensor, optional) – 潜在空间用于 \(\log\sigma\)。如果设置为 None，则使用最后一次计算的 \(\log\sigma^2\)。（默认值：None）

Return type:

Tensor