From 5961c140265ec5b9df9dd3bfed98f96f8c1d7bde Mon Sep 17 00:00:00 2001 From: Tomohiro Endo Date: Wed, 23 Jun 2021 16:49:31 +0900 Subject: [PATCH] [Doc] EdgeWeightNorm Fix doc (#3051) Co-authored-by: Jinjing Zhou --- python/dgl/nn/pytorch/conv/graphconv.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/dgl/nn/pytorch/conv/graphconv.py b/python/dgl/nn/pytorch/conv/graphconv.py index ec75e155dab0..21a01a5c7103 100644 --- a/python/dgl/nn/pytorch/conv/graphconv.py +++ b/python/dgl/nn/pytorch/conv/graphconv.py @@ -21,12 +21,12 @@ class EdgeWeightNorm(nn.Module): Mathematically, setting ``norm='both'`` yields the following normalization term: - .. math: + .. math:: c_{ji} = (\sqrt{\sum_{k\in\mathcal{N}(j)}e_{jk}}\sqrt{\sum_{k\in\mathcal{N}(i)}e_{ki}}) And, setting ``norm='right'`` yields the following normalization term: - .. math: + .. math:: c_{ji} = (\sum_{k\in\mathcal{N}(i)}}e_{ki}) where :math:`e_{ji}` is the scalar weight on the edge from node :math:`j` to node :math:`i`.