conv3d源码 pytorch_PyTorch Geometric(PyG) torch_geometric.nn.conv.gcn_conv源码详解

import torch

from torch.nn import Parameter

from torch_scatter import scatter_add

from torch_geometric.nn.conv import MessagePassing

from torch_geometric.utils import add_remaining_self_loops

from ..inits import glorot, zeros

class GCNConv(MessagePassing):

r"""The graph convolutional operator from the `"Semi-supervised

Classification with Graph Convolutional Networks"

`_ paper

.. math::

\mathbf{X}^{\prime} = \mathbf{\hat{D}}^{-1/2} \mathbf{\hat{A}}

\mathbf{\hat{D}}^{-1/2} \mathbf{X} \mathbf{\Theta},

where :math:`\mathbf{\hat{A}} = \mathbf{A} + \mathbf{I}` denotes the

adjacency matrix with inserted self-loops and

:math:`\hat{D}_{ii} = \sum_{j=0} \hat{A}_{ij}` its diagonal degree matrix.

Args:

in_channels (int): Size of each input sample.</