mindspore_gl.nn

APIs for graph convolutions.

API Name

Description

Supported Platforms

mindspore_gl.nn.AGNNConv

Attention Based Graph Neural Network.

Ascend GPU

mindspore_gl.nn.APPNPConv

Approximate Personalization Propagation in Neural Prediction Layers.

Ascend GPU

mindspore_gl.nn.ASTGCN

Attention Based Spatial-Temporal Graph Convolutional Networks.

Ascend GPU

mindspore_gl.nn.AvgPooling

Apply average pooling to the nodes in the batched graph.

Ascend GPU

mindspore_gl.nn.CFConv

CFConv in SchNet.

Ascend GPU

mindspore_gl.nn.ChebConv

Chebyshev Spectral Graph Convolution layer.

Ascend GPU

mindspore_gl.nn.DOTGATConv

Applying a dot product version of self-attention in GAT.

Ascend GPU

mindspore_gl.nn.EDGEConv

EdgeConv layer.

Ascend GPU

mindspore_gl.nn.EGConv

Efficient Graph Convolution.

Ascend GPU

mindspore_gl.nn.GATConv

Graph Attention Network.

Ascend GPU

mindspore_gl.nn.GatedGraphConv

Gated Graph Convolution Layer.

Ascend GPU

mindspore_gl.nn.GATv2Conv

Graph Attention Network v2.

Ascend GPU

mindspore_gl.nn.GCNConv

Graph Convolution Network Layer.

Ascend GPU

mindspore_gl.nn.GCNConv2

Graph Convolution Network Layer.

Ascend GPU

mindspore_gl.nn.GINConv

Graph isomorphic network layer.

Ascend GPU

mindspore_gl.nn.GlobalAttentionPooling

Apply global attention pooling to the nodes in the graph.

Ascend GPU

mindspore_gl.nn.GMMConv

Gaussian mixture model convolutional layer.

Ascend GPU

mindspore_gl.nn.GNNCell

GNN Cell class.

Ascend GPU

mindspore_gl.nn.MaxPooling

Apply maximum pooling to the nodes in the batched graph.

Ascend GPU

mindspore_gl.nn.MeanConv

GraphSAGE Layer.

Ascend GPU

mindspore_gl.nn.NNConv

Graph convolutional layer.

Ascend GPU

mindspore_gl.nn.SAGEConv

GraphSAGE Layer.

Ascend GPU

mindspore_gl.nn.SAGPooling

The self-attention pooling operator.

Ascend GPU

mindspore_gl.nn.Set2Set

Sequence to sequence for sets.

Ascend GPU

mindspore_gl.nn.SGConv

Simplified Graph convolutional layer.

Ascend GPU

mindspore_gl.nn.SortPooling

Apply sort pooling to the nodes in the graph.

Ascend GPU

mindspore_gl.nn.STConv

Spatial-Temporal Graph Convolutional layer.

Ascend GPU

mindspore_gl.nn.SumPooling

Apply sum pooling to the nodes in the batched graph.

Ascend GPU

mindspore_gl.nn.TAGConv

Topology adaptation graph convolutional layer.

Ascend GPU

mindspore_gl.nn.WeightAndSum

Calculates importance weights for nodes and performs weighted sums.

Ascend GPU