gigl.src.common.models.layers.l2_normalize_embeddings

gigl.src.common.models.layers.normalization.l2_normalize_embeddings(node_typed_embeddings: Tensor | Dict[NodeType, Tensor]) → Tensor | Dict[NodeType, Tensor]