MCPcopy Create free account

hub / github.com/ZZy979/pytorch-tutorial / functions

Functions479 in github.com/ZZy979/pytorch-tutorial

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
Methodapply_edges
(self, edges)
gnn/dgl/model.py:130
Methoddownload
(self)
gnn/data/acm.py:48
Methoddownload
(self)
gnn/data/acm.py:169
Methoddownload
(self)
gnn/data/dblp.py:55
Methoddownload
(self)
gnn/data/dblp.py:226
Methoddownload
(self)
gnn/data/aminer.py:46
Methoddownload
(self)
gnn/data/aminer.py:176
Methoddownload
(self)
gnn/data/heco.py:40
Methoddownload
(self)
gnn/data/imdb.py:53
Methoddownload
(self)
gnn/data/imdb.py:184
Functiondraw
(i, g, ax, all_logits)
gnn/dgl/dgl_first_demo.py:39
Methodfix_input
(y)
gnn/cs/model.py:111
Methodforward
: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
Methodforward
:param pair_graph: DGLGraph 用户-物品子图 :param blocks: List[DGLBlock] 知识图谱的MFG,blocks[-1].dstnodes()对应items :return: tensor(B) 用户
kgrec/kgcn/model.py:66
Methodforward
:param x: tensor(*, 28, 28) 输入图像 :return: tensor(*, 10)
dlwizard/cnn.py:19
Methodforward
(self, x)
dlwizard/fnn.py:16
Methodforward
(self, x)
dlwizard/logistic_regression.py:14
Methodforward
(self, x)
dlwizard/linear_regression.py:15
Methodforward
:param x: tensor(batch, seq_len, d_in) :return: tensor(batch, d_out)
dlwizard/rnn.py:18
Methodforward
:param x: tensor(batch, seq_len, d_in) :return: tensor(batch, d_out)
dlwizard/lstm.py:18
Methodforward
(self, x)
beginner/cifar10_tutorial.py:45
Methodforward
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
Methodforward
(self, x)
beginner/neural_networks_tutorial.py:19
Methodforward
:param feats: tensor(N, C, d_in) 输入特征列表,N为batch大小,C为输入特征个数 :return: tensor(N, d_hid) 顶点的异构内容嵌入向量
gnn/hetgnn/model.py:22
Methodforward
:param embeds: tensor(N, Nt, d) 邻居的内容嵌入,N为batch大小,Nt为每个顶点的邻居个数 :return: tensor(N, d) 顶点的t类型邻居聚集嵌入
gnn/hetgnn/model.py:41
Methodforward
: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
Methodforward
: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
Methodforward
:param g: DGLGraph 同构图 :param h: tensor(N, d_in) 输入特征,N为g的顶点数 :return: tensor(N, d_out) 输出顶点特征,K为注意力头数
gnn/gat/model.py:44
Methodforward
(self, g, h)
gnn/dgl/graph_clf.py:21
Methodforward
(self, edge_subgraph, blocks, x, etype)
gnn/dgl/edge_clf_hetero_mb.py:21
Methodforward
(self, pos_g, neg_g, blocks, x, etype)
gnn/dgl/link_pred_hetero_mb.py:21
Methodforward
(self, g, x, dec_graph)
gnn/dgl/edge_type_hetero.py:21
Methodforward
(self, edge_subgraph, blocks, x)
gnn/dgl/edge_clf_mb.py:22
Methodforward
(self, g, neg_g, x, etype)
gnn/dgl/link_pred_hetero.py:32
Methodforward
(self, pos_g, neg_g, blocks, x)
gnn/dgl/link_pred_mb.py:22
Methodforward
(self, g, x, etype)
gnn/dgl/edge_clf_hetero.py:20
Methodforward
(self, g, neg_g, x)
gnn/dgl/link_pred.py:28
Methodforward
(self, g, x)
gnn/dgl/edge_clf.py:21
Methodforward
(self, g)
gnn/dgl/graph_clf_hetero.py:25
Methodforward
(self, g, inputs)
gnn/dgl/model.py:15
Methodforward
(self, blocks, inputs)
gnn/dgl/model.py:29
Methodforward
(self, g, inputs)
gnn/dgl/model.py:43
Methodforward
(self, g, inputs)
gnn/dgl/model.py:61
Methodforward
(self, blocks, inputs)
gnn/dgl/model.py:79
Methodforward
(self, graph, h)
gnn/dgl/model.py:88
Methodforward
(self, graph, h)
gnn/dgl/model.py:106
Methodforward
(self, graph, h, etype)
gnn/dgl/model.py:116
Methodforward
(self, graph, h, etype)
gnn/dgl/model.py:134
Methodforward
(self, pos_score, neg_score)
gnn/dgl/model.py:144
Methodforward
: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
Methodforward
: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
Methodforward
返回ΔT对应的相对时间编码
gnn/hgt/model.py:160
Methodforward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in)] 顶点类型到输入顶点特征的映射 :return: tensor(N_i, d_out) 待预测顶点的最终嵌入
gnn/hgt/model.py:194
Methodforward
:param z: tensor(N, M, d_in) 顶点基于不同元路径的嵌入,N为顶点数,M为元路径个数 :return: tensor(N, d_in) 顶点的最终嵌入
gnn/han/model.py:27
Methodforward
:param gs: List[DGLGraph] 基于元路径的邻居组成的同构图 :param h: tensor(N, d_in) 输入顶点特征 :return: tensor(N, K*d_out) 输出顶点特征
gnn/han/model.py:59
Methodforward
:param gs: List[DGLGraph] 基于元路径的邻居组成的同构图 :param h: tensor(N, d_in) 输入顶点特征 :return: tensor(N, d_out) 输出顶点嵌入
gnn/han/model.py:87
Methodforward
: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
Methodforward
:param g: DGLGraph 异构图 :return: Dict[str, tensor(N_i, d_out)] 顶点类型到顶点嵌入的映射
gnn/rgcn/model_hetero.py:109
Methodforward
:param embed: tensor(N, d) 实体嵌入 :param head: tensor(*) 头实体 :param rel: tensor(*) 关系 :param tail: tensor(*) 尾实体
gnn/rgcn/model.py:26
Methodforward
:param g: DGLGraph 同构图 :param etypes: tensor(|E|) 边类型 :return: tensor(N, d_hid) 顶点嵌入
gnn/rgcn/model.py:64
Methodforward
:param g: DGLGraph 无向图 :param labels: tensor(N) 标签 :param mask: tensor(N), optional 有标签顶点mask :return: tensor(N, C) 预
gnn/lp/model.py:27
Methodforward
: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
Methodforward
: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
Methodforward
: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
Methodforward
:param g: DGLGraph 异构图 :param feats: Dict[str, tensor(N_i, d_in_i)] 顶点类型到输入顶点特征的映射 :return: tensor(N_i, d_out) 待预测顶点的最终嵌入
gnn/hgconv/model.py:248
Methodforward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:40
Methodforward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:52
Methodforward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:63
Methodforward
(self, g, feat_src, feat_dst)
gnn/supergat/attention.py:69
Methodforward
:param g: DGLGraph 同构图 :param feat: tensor(N_src, d_in) 输入顶点特征 :return: tensor(N_dst, K, d_out) 输出顶点特征
gnn/supergat/model.py:44
Methodforward
:param g: DGLGraph 同构图 :param feat: tensor(N, d_in) 输入顶点特征 :return: tensor(N, d_out), tensor(1) 输出顶点特征和自监督注意力损失
gnn/supergat/model.py:110
Methodforward
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
Methodforward
(self, x)
gnn/cs/model.py:26
Methodforward
:param g: DGLGraph 无向图 :param labels: tensor(N, C) one-hot标签 :param mask: tensor(N), optional 有标签顶点mask :param post_s
gnn/cs/model.py:55
Methodforward
:param g: DGLGraph 无向图 :param labels: tensor(N, C) one-hot标签 :param base_pred: tensor(N, C) 基础预测 :param mask: tensor(
gnn/cs/model.py:133
Methodforward
: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
Methodforward
:param feats: tensor(N_R, N, K*d) dtype类型顶点在不同关系下的表示 :return: tensor(N, K*d) 跨关系消息传递后dtype类型顶点在关系R下的表示
gnn/rhgnn/model.py:95
Methodforward
:param node_feats: Dict[str, tensor(N, K*d_node)] 边类型到顶点在该关系下的表示的映射 :param rel_feats: Dict[str, tensor(K*d_rel)] 边类型到关系的表示的映射
gnn/rhgnn/model.py:137
Methodforward
:param g: DGLGraph 异构图 :param feats: Dict[(str, str, str), tensor(N_i, d_in)] 关系(三元组)到目标顶点输入特征的映射 :param rel_feats: Dict[str,
gnn/rhgnn/model.py:239
Methodforward
: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
Methodforward
:param x: tensor(N, d_in) :return: tensor(N, d_out)
gnn/sign/model.py:25
Methodforward
:param feats: List[tensor(N, d_in)] 每一跳的邻居聚集特征,长度为r+1 :return: tensor(N, d_out) 输出顶点特征
gnn/sign/model.py:60
Methodforward
:param feat: tensor(E, L, d_in) :return: tensor(E, d_out)
gnn/magnn/encoder.py:7
Methodforward
(self, feat)
gnn/magnn/encoder.py:20
Methodforward
(self, feat)
gnn/magnn/encoder.py:30
Methodforward
:param g: DGLGraph 基于给定元路径的邻居组成的图,每条边表示一个元路径实例 :param node_feat: tensor(N, d_in) 输入顶点特征,N为g的终点个数 :param edge_feat: tensor(E,
gnn/magnn/model.py:47
Methodforward
:param z: tensor(N, M, d_in) 每个顶点关于所有元路径的嵌入,N为顶点数,M为元路径个数 :return: tensor(N, d_in) 聚集后的顶点嵌入
gnn/magnn/model.py:89
← previousnext →301–400 of 479, ranked by callers