Toro

GraphUtils

FunctionDescription
addSelfLoops edgeIndex numNodesAdd self-loops to edge index. Returns (newEdgeIndex, numNodes).
degree index numNodes dtype deviceCompute node degree from an index vector (one row of edge_index).

addSelfLoops

addSelfLoops edgeIndex numNodes

Add self-loops to edge index. Returns (newEdgeIndex, numNodes).

Parameters

  • edgeIndex : Tensor
  • numNodes : int64

Returns Tensor

degree

degree index numNodes dtype device

Compute node degree from an index vector (one row of edge_index).

Parameters

  • index : Tensor
  • numNodes : int64
  • dtype : ScalarType
  • device : Device

Returns Tensor