dgl.ops.copy_u_min

dgl.ops.copy_u_min(g, x)

Generalized SpMM function. It copies node feature to edge as the message. Then aggregates the message by min on destination nodes.

Parameters
  • g (DGLHeteroGraph) – The input graph

  • x (tensor) – The source node features.

Returns

The result tensor.

Return type

tensor

Notes

This function supports autograd (computing input gradients given the output gradient).

The min function will return zero for nodes with no incoming messages. This is implemented by replacing all infinity values to zero.