I want to create a graph attention network whose nodes are the feature matrix and whose edges are the similarity matrix. I already have the feature matrix and the similarity matrix.
How can I implement the code?

The dimensions of the matrices:

train_feature_matrix= torch.Size([2939, 25])

train_similarity_matrix= torch.Size([2939, 2939])