gigl.src.common.models.layers.normalization#

Functions#

l2_normalize_embeddings(…)

Module Contents#

gigl.src.common.models.layers.normalization.l2_normalize_embeddings(node_typed_embeddings: torch.Tensor) → torch.Tensor[source]#
gigl.src.common.models.layers.normalization.l2_normalize_embeddings(node_typed_embeddings: dict[gigl.src.common.types.graph_data.NodeType, torch.Tensor]) → dict[gigl.src.common.types.graph_data.NodeType, torch.Tensor]