EdgeWeighting

class EdgeWeighting[source]

Bases: torch.nn.modules.module.Module

Base class for edge weightings.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

Methods Summary

forward(source, target)

Compute edge weights.

Methods Documentation

abstract forward(source, target)[source]

Compute edge weights.

Parameters

  • source (LongTensor) – shape: (num_edges,) The source indices.

  • target (LongTensor) – shape: (num_edges,) The target indices.

Return type

FloatTensor

Returns

shape: (num_edges,) The edge weights.