utils
¶
Module Contents¶
Functions¶
|
|
|
|
|
|
|
|
|
Args: |
|
|
|
Args: |
|
Args: |
|
-
utils.
spmm
(indices, values, b)[source]¶ Args: indices : Tensor, shape=(2, E) values : Tensor, shape=(E,) shape : tuple(int ,int) b : Tensor, shape=(N, )
-
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