Method encode

src/ml_models/graph_networks/gnn_head.py:134–135

(self)

        return torch.sum(token_embeddings * input_mask_expanded, 1) / torch.clamp(input_mask_expanded.sum(1), min=1e-9)

    def encode(self):
        pass


class GATLayer(nn.Module):