spamosaic.architectures.gat.GAT

class spamosaic.architectures.gat.GAT(*args: Any, **kwargs: Any)

A 4-layer Graph Attention Network used for spatial transcriptomics.

This module implements a symmetric encoder-decoder GAT with tied weights.

Parameters:

hidden_dims (list of int) – A list [in_dim, hidden_dim, out_dim] specifying the feature dimensions.