MCPcopy Create free account

hub / github.com/cszhangzhen/HGP-SL / functions

Functions27 in github.com/cszhangzhen/HGP-SL

↓ 2 callersMethod__init__
(self, in_channels, out_channels, cached=False, bias=True, **kwargs)
layers.py:42
↓ 2 callersFunctioncompute_test
(loader)
main.py:111
↓ 1 callersFunction_make_ix_like
(batch)
sparse_softmax.py:53
↓ 1 callersFunction_threshold_and_support
Sparsemax building block: compute the threshold Args: x: input tensor to apply the sparsemax batch: group indicators Returns:
sparse_softmax.py:61
↓ 1 callersMethodbackward
(ctx, grad_output)
sparse_softmax.py:109
↓ 1 callersMethodnorm
(edge_index, num_nodes, edge_weight, dtype=None)
layers.py:67
↓ 1 callersMethodnorm
(edge_index, num_nodes, edge_weight, dtype=None)
layers.py:117
↓ 1 callersMethodreset_parameters
(self)
layers.py:62
↓ 1 callersFunctionscatter_sort
(x, batch, fill_value=-1e16)
sparse_softmax.py:27
↓ 1 callersFunctiontrain
()
main.py:56
Method__call__
(self, data)
layers.py:15
Method__init__
(self, improved=False, cached=False, **kwargs)
layers.py:108
Method__init__
(self, in_channels, ratio=0.8, sample=False, sparse=False, sl=True, lamb=1.0, negative_slop=0.2)
layers.py:157
Method__init__
(self, args)
models.py:10
Method__init__
(self)
sparse_softmax.py:125
Method__repr__
(self)
layers.py:37
Method__repr__
(self)
layers.py:103
Methodforward
(self, x, edge_index, edge_weight=None)
layers.py:78
Methodforward
(self, x, edge_index, edge_weight)
layers.py:134
Methodforward
(self, x, edge_index, edge_attr, batch=None)
layers.py:173
Methodforward
(self, data)
models.py:34
Methodforward
sparsemax: normalizing sparse transform Parameters: ctx: context object x (Tensor): shape (N, ) batch: gro
sparse_softmax.py:91
Methodforward
(self, x, batch)
sparse_softmax.py:128
Methodmessage
(self, x_j, norm)
layers.py:95
Methodmessage
(self, x_j, norm)
layers.py:149
Methodupdate
(self, aggr_out)
layers.py:98
Methodupdate
(self, aggr_out)
layers.py:152