Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ZZy979/pytorch-tutorial
/ functions
Functions
479 in github.com/ZZy979/pytorch-tutorial
⨍
Functions
479
◇
Types & classes
107
Method
__len__
(self)
gnn/data/dblp.py:184
Method
__len__
(self)
gnn/data/dblp.py:270
Method
__len__
(self)
gnn/data/dgl.py:90
Method
__len__
(self)
gnn/data/dgl.py:120
Method
__len__
(self)
gnn/data/aminer.py:132
Method
__len__
(self)
gnn/data/aminer.py:269
Method
__len__
(self)
gnn/data/heco.py:114
Method
__len__
(self)
gnn/data/imdb.py:143
Method
__len__
(self)
gnn/data/imdb.py:230
Method
_collate
根据边id采样子图 :param items: tensor(B) 边id :return: tensor(N_src), DGLGraph, List[DGLBlock] 知识图谱的输入顶点id,用户-物品图的边子图, 知识图谱根据边id关联的物品
kgrec/kgcn/dataloader.py:20
Method
_collate_with_negative_sampling
根据边id采样子图,并进行负采样 :param items: tensor(B) 边id :return: tensor(N_src), DGLGraph, DGLGraph, List[DGLBlock] 知识图谱的输入顶点id,用户-物品图的边子图,负样本图,
kgrec/kgcn/dataloader.py:34
Method
_generate
(self, g, eids, canonical_etype)
gnn/utils/neg_sampler.py:18
Method
_read_feats
(self)
gnn/data/heco.py:197
Method
apply_edges
(self, edges)
gnn/dgl/model.py:130
Method
download
(self)
gnn/data/acm.py:48
Method
download
(self)
gnn/data/acm.py:169
Method
download
(self)
gnn/data/dblp.py:55
Method
download
(self)
gnn/data/dblp.py:226
Method
download
(self)
gnn/data/aminer.py:46
Method
download
(self)
gnn/data/aminer.py:176
Method
download
(self)
gnn/data/heco.py:40
Method
download
(self)
gnn/data/imdb.py:53
Method
download
(self)
gnn/data/imdb.py:184
Function
draw
(i, g, ax, all_logits)
gnn/dgl/dgl_first_demo.py:39
Method
fix_input
(y)
gnn/cs/model.py:111
Method
forward
:param src_feat: tensor(B, K^h, K, d) 输入实体表示,B为batch大小,K为邻居个数,h为跳步数/层数 :param dst_feat: tensor(B, K^h, d) 目标实体表示 :param rel_f
kgrec/kgcn/model.py:27
Method
forward
:param pair_graph: DGLGraph 用户-物品子图 :param blocks: List[DGLBlock] 知识图谱的MFG,blocks[-1].dstnodes()对应items :return: tensor(B) 用户
kgrec/kgcn/model.py:66
Method
forward
:param x: tensor(*, 28, 28) 输入图像 :return: tensor(*, 10)
dlwizard/cnn.py:19
Method
forward
(self, x)
dlwizard/fnn.py:16
Method
forward
(self, x)
dlwizard/logistic_regression.py:14
Method
forward
(self, x)
dlwizard/linear_regression.py:15
Method
forward
:param x: tensor(batch, seq_len, d_in) :return: tensor(batch, d_out)
dlwizard/rnn.py:18
Method
forward
:param x: tensor(batch, seq_len, d_in) :return: tensor(batch, d_out)
dlwizard/lstm.py:18
Method
forward
(self, x)
beginner/cifar10_tutorial.py:45
Method
forward
In the forward function we accept a Tensor of input data and we must return a Tensor of output data. We can use Modules defined in th
beginner/two_layer_net.py:19
Method
forward
(self, x)
beginner/neural_networks_tutorial.py:19
Method
forward
:param feats: tensor(N, C, d_in) 输入特征列表,N为batch大小,C为输入特征个数 :return: tensor(N, d_hid) 顶点的异构内容嵌入向量
gnn/hetgnn/model.py:22
Method
forward
:param embeds: tensor(N, Nt, d) 邻居的内容嵌入,N为batch大小,Nt为每个顶点的邻居个数 :return: tensor(N, d) 顶点的t类型邻居聚集嵌入
gnn/hetgnn/model.py:41
Method
forward
:param content_embed: tensor(N, d) 内容嵌入,N为batch大小 :param neighbor_embeds: tensor(A, N, d) 邻居嵌入,A为邻居类型数 :return: tensor(N, d)
gnn/hetgnn/model.py:61
Method
forward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, C_i, d_in)] 顶点类型到输入特征的映射 :return: Dict[str, tensor(N_i, d_hid)] 顶点
gnn/hetgnn/model.py:99
Method
forward
:param g: DGLGraph 同构图 :param h: tensor(N, d_in) 输入特征,N为g的顶点数 :return: tensor(N, d_out) 输出顶点特征,K为注意力头数
gnn/gat/model.py:44
Method
forward
(self, g, h)
gnn/dgl/graph_clf.py:21
Method
forward
(self, edge_subgraph, blocks, x, etype)
gnn/dgl/edge_clf_hetero_mb.py:21
Method
forward
(self, pos_g, neg_g, blocks, x, etype)
gnn/dgl/link_pred_hetero_mb.py:21
Method
forward
(self, g, x, dec_graph)
gnn/dgl/edge_type_hetero.py:21
Method
forward
(self, edge_subgraph, blocks, x)
gnn/dgl/edge_clf_mb.py:22
Method
forward
(self, g, neg_g, x, etype)
gnn/dgl/link_pred_hetero.py:32
Method
forward
(self, pos_g, neg_g, blocks, x)
gnn/dgl/link_pred_mb.py:22
Method
forward
(self, g, x, etype)
gnn/dgl/edge_clf_hetero.py:20
Method
forward
(self, g, neg_g, x)
gnn/dgl/link_pred.py:28
Method
forward
(self, g, x)
gnn/dgl/edge_clf.py:21
Method
forward
(self, g)
gnn/dgl/graph_clf_hetero.py:25
Method
forward
(self, g, inputs)
gnn/dgl/model.py:15
Method
forward
(self, blocks, inputs)
gnn/dgl/model.py:29
Method
forward
(self, g, inputs)
gnn/dgl/model.py:43
Method
forward
(self, g, inputs)
gnn/dgl/model.py:61
Method
forward
(self, blocks, inputs)
gnn/dgl/model.py:79
Method
forward
(self, graph, h)
gnn/dgl/model.py:88
Method
forward
(self, graph, h)
gnn/dgl/model.py:106
Method
forward
(self, graph, h, etype)
gnn/dgl/model.py:116
Method
forward
(self, graph, h, etype)
gnn/dgl/model.py:134
Method
forward
(self, pos_score, neg_score)
gnn/dgl/model.py:144
Method
forward
:param g: DGLGraph 二分图(只包含一种关系) :param feat: tensor(N_src, d_in) or (tensor(N_src, d_in), tensor(N_dst, d_in)) 输入特征 :return:
gnn/hgt/model.py:41
Method
forward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in)] 顶点类型到输入顶点特征的映射 :return: Dict[str, tensor(N_i, d_out)] 顶点类型到
gnn/hgt/model.py:112
Method
forward
返回ΔT对应的相对时间编码
gnn/hgt/model.py:160
Method
forward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in)] 顶点类型到输入顶点特征的映射 :return: tensor(N_i, d_out) 待预测顶点的最终嵌入
gnn/hgt/model.py:194
Method
forward
:param z: tensor(N, M, d_in) 顶点基于不同元路径的嵌入,N为顶点数,M为元路径个数 :return: tensor(N, d_in) 顶点的最终嵌入
gnn/han/model.py:27
Method
forward
:param gs: List[DGLGraph] 基于元路径的邻居组成的同构图 :param h: tensor(N, d_in) 输入顶点特征 :return: tensor(N, K*d_out) 输出顶点特征
gnn/han/model.py:59
Method
forward
:param gs: List[DGLGraph] 基于元路径的邻居组成的同构图 :param h: tensor(N, d_in) 输入顶点特征 :return: tensor(N, d_out) 输出顶点嵌入
gnn/han/model.py:87
Method
forward
:param g: DGLGraph 异构图 :param inputs: Dict[str, tensor(N_i, d_in)] 顶点类型到输入特征的映射 :return: Dict[str, tensor(N_i, d_out)] 顶点类型到输
gnn/rgcn/model_hetero.py:53
Method
forward
:param g: DGLGraph 异构图 :return: Dict[str, tensor(N_i, d_out)] 顶点类型到顶点嵌入的映射
gnn/rgcn/model_hetero.py:109
Method
forward
:param embed: tensor(N, d) 实体嵌入 :param head: tensor(*) 头实体 :param rel: tensor(*) 关系 :param tail: tensor(*) 尾实体
gnn/rgcn/model.py:26
Method
forward
:param g: DGLGraph 同构图 :param etypes: tensor(|E|) 边类型 :return: tensor(N, d_hid) 顶点嵌入
gnn/rgcn/model.py:64
Method
forward
:param g: DGLGraph 无向图 :param labels: tensor(N) 标签 :param mask: tensor(N), optional 有标签顶点mask :return: tensor(N, C) 预
gnn/lp/model.py:27
Method
forward
:param g: DGLGraph 二分图(只包含一种关系) :param feat: tensor(N_src, d_in) or (tensor(N_src, d_in), tensor(N_dst, d_in)) 输入特征 :return:
gnn/hgconv/model.py:42
Method
forward
:param node_feats: Dict[str, tensor(N_i, d_in) 顶点类型到输入顶点特征的映射 :param rel_feats: Dict[(str, str, str), tensor(N_i, K*d_out)]
gnn/hgconv/model.py:94
Method
forward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in)] 顶点类型到输入顶点特征的映射 :return: Dict[str, tensor(N_i, K*d_out)] 顶点类
gnn/hgconv/model.py:191
Method
forward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in_i)] 顶点类型到输入顶点特征的映射 :return: tensor(N_i, d_out) 待预测顶点的最终嵌入
gnn/hgconv/model.py:248
Method
forward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:40
Method
forward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:52
Method
forward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:63
Method
forward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:69
Method
forward
:param g: DGLGraph 同构图 :param feat: tensor(N_src, d_in) 输入顶点特征 :return: tensor(N_dst, K, d_out) 输出顶点特征
gnn/supergat/model.py:44
Method
forward
:param g: DGLGraph 同构图 :param feat: tensor(N, d_in) 输入顶点特征 :return: tensor(N, d_out), tensor(1) 输出顶点特征和自监督注意力损失
gnn/supergat/model.py:110
Method
forward
r"""给定中心词、正样本和负样本,返回似然函数的相反数(损失): .. math:: L=-\log {\sigma(v_c \cdot v_p)}-\sum_{n \in neg}{\log {\sigma(-v_c \cdot v_n)}}
gnn/metapath2vec/skipgram.py:13
Method
forward
(self, x)
gnn/cs/model.py:26
Method
forward
:param g: DGLGraph 无向图 :param labels: tensor(N, C) one-hot标签 :param mask: tensor(N), optional 有标签顶点mask :param post_s
gnn/cs/model.py:55
Method
forward
:param g: DGLGraph 无向图 :param labels: tensor(N, C) one-hot标签 :param base_pred: tensor(N, C) 基础预测 :param mask: tensor(
gnn/cs/model.py:133
Method
forward
:param g: DGLGraph 二分图(只包含一种关系) :param feat: tensor(N_src, d_in) or (tensor(N_src, d_in), tensor(N_dst, d_in)) 输入特征 :param fe
gnn/rhgnn/model.py:45
Method
forward
:param feats: tensor(N_R, N, K*d) dtype类型顶点在不同关系下的表示 :return: tensor(N, K*d) 跨关系消息传递后dtype类型顶点在关系R下的表示
gnn/rhgnn/model.py:95
Method
forward
:param node_feats: Dict[str, tensor(N, K*d_node)] 边类型到顶点在该关系下的表示的映射 :param rel_feats: Dict[str, tensor(K*d_rel)] 边类型到关系的表示的映射
gnn/rhgnn/model.py:137
Method
forward
:param g: DGLGraph 异构图 :param feats: Dict[(str, str, str), tensor(N_i, d_in)] 关系(三元组)到目标顶点输入特征的映射 :param rel_feats: Dict[str,
gnn/rhgnn/model.py:239
Method
forward
:param blocks: blocks: List[DGLBlock] :param feats: Dict[str, tensor(N_i, d_in_i)] 顶点类型到输入顶点特征的映射 :return: tensor(N_i, d_out)
gnn/rhgnn/model.py:350
Method
forward
:param x: tensor(N, d_in) :return: tensor(N, d_out)
gnn/sign/model.py:25
Method
forward
:param feats: List[tensor(N, d_in)] 每一跳的邻居聚集特征,长度为r+1 :return: tensor(N, d_out) 输出顶点特征
gnn/sign/model.py:60
Method
forward
:param feat: tensor(E, L, d_in) :return: tensor(E, d_out)
gnn/magnn/encoder.py:7
Method
forward
(self, feat)
gnn/magnn/encoder.py:20
Method
forward
(self, feat)
gnn/magnn/encoder.py:30
Method
forward
:param g: DGLGraph 基于给定元路径的邻居组成的图,每条边表示一个元路径实例 :param node_feat: tensor(N, d_in) 输入顶点特征,N为g的终点个数 :param edge_feat: tensor(E,
gnn/magnn/model.py:47
Method
forward
:param z: tensor(N, M, d_in) 每个顶点关于所有元路径的嵌入,N为顶点数,M为元路径个数 :return: tensor(N, d_in) 聚集后的顶点嵌入
gnn/magnn/model.py:89
← previous
next →
301–400 of 479, ranked by callers