torch_geometric

https://pytorch-geometric.readthedocs.io/en/latest/modules/nn.html#torch_geometric.nn.conv.SAGEConv

1.torch_geometric.utils



1.1

在这里插入图片描述

1.2 to_dense_batch

在这里插入图片描述

这里的to_dense_batch 不是把X变得更稀疏了吗?
X ∈ R B × N max ⁡ × F \mathbf{X} \in \mathbb{R}^{B \times N_{\max} \times F} XRB×Nmax×F第一维度不是比 X ∈ R ( N 1 + … + N B ) × F \mathbf{X} \in \mathbb{R}^{(N_1 + \ldots + N_B) \times F} XR(N1++NB)×F更大了




   

2.torch_geometric.nn

2.1 DenseSAGEConv

只适用于二值权重(0、1)矩阵,如果带权重的话要使用 torch_geometric.nn.dense.DenseGraphConv

class DenseSAGEConv(in_channels, out_channels, normalize=False, bias=True)

The adjacency tensor is broadcastable in the batch dimension, resulting in a shared adjacency matrix for the complete batch.

forward(x, adj, mask=None)

forward中的Args:

  1. x x x (Tensor) -节点特征tensor
    X ∈ R B × N × F \mathbf{X} \in \mathbb{R}^{B \times N \times F} XRB×N×F
    batch-size B B B,
    (maximum) number of nodes N N Nfor each graph,
    feature dimension F F F.

  2. a d j adj adj (Tensor): Adjacency tensor : A ∈ R B × N × N \mathbf{A} \in \mathbb{R}^{B \times N \times N} ARB×N×N. The adjacency tensor is broadcastable in
    the batch dimension, resulting in a shared adjacency matrix for
    the complete ba

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值