COSIE.model_component.GraphAutoencoder.decoder

GraphAutoencoder.decoder(x, edge_index)[source]

Decode latent embeddings to reconstruct original node features.

Parameters

xtorch.Tensor: Latent node embeddings of shape (n_nodes, latent_dim).
edge_indextorch.Tensor: Edge index defining the graph connectivity.

Returns

xtorch.Tensor: Reconstructed node features of shape (n_nodes, in_dim).