gammagl.utils.degree

class degree(index, num_nodes: int | None = None, dtype=None)[source]

Computes the (unweighted) degree of a given one-dimensional index tensor.

Parameters:

  • index (tensor) – Index tensor.

  • num_nodes (int, optional) – The number of nodes, i.e. max_val + 1 of index. (default: None)

  • dtype (tlx.dtype, optional) – The desired data type of the returned tensor.

Return type:

Tensor