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

Functions#

l2_normalize_embeddings(node_typed_embeddings)

Module Contents#

gigl.src.common.models.layers.normalization.l2_normalize_embeddings(node_typed_embeddings)[source]#
Parameters:

node_typed_embeddings (Union[torch.Tensor, Dict[gigl.src.common.types.graph_data.NodeType, torch.Tensor]])

Return type:

Union[torch.Tensor, Dict[gigl.src.common.types.graph_data.NodeType, torch.Tensor]]