COSIE.model_component.GraphAutoencoder.forward

GraphAutoencoder.forward(x, edge_index)

Perform full encoder-decoder forward pass.

Parameters

xtorch.Tensor

Input node feature matrix of shape (n_nodes, in_dim).

edge_indextorch.Tensor

Edge index defining the graph connectivity.

Returns

x_hattorch.Tensor

Reconstructed node feature matrix of shape (n_nodes, in_dim).