cogdl.utils
¶
Module Contents¶
Functions¶
|
|
|
|
|
|
|
|
|
|
|
Args: |
|
|
|
|
|
Args: |
|
Args: |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
-
cogdl.utils.
add_remaining_self_loops
(edge_index, edge_weight=None, fill_value=1, num_nodes=None)[source]¶
-
cogdl.utils.
spmm
(indices, values, b)[source]¶ Args: indices : Tensor, shape=(2, E) values : Tensor, shape=(E,) shape : tuple(int ,int) b : Tensor, shape=(N, )
-
cogdl.utils.
edge_softmax
(indices, values, shape)[source]¶ - Args:
indices: Tensor, shape=(2, E) values: Tensor, shape=(N,) shape: tuple(int, int)
- Returns:
Softmax values of edge values for nodes