COSIE.model_component.GraphAutoencoder.encoder

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

Encode input node features into latent embeddings.

Parameters

xtorch.Tensor: Input node feature matrix of shape (n_nodes, in_dim).
edge_indextorch.Tensor: Edge index defining the graph connectivity.

Returns

xtorch.Tensor: Latent node embeddings of shape (n_nodes, latent_dim).