SymmetricEdgeWeighting

class SymmetricEdgeWeighting

Bases: pykeen.nn.weighting.EdgeWeighting

Normalize messages by product of inverse sqrt of in-degree and out-degree.

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

Methods Summary

forward(source, target)

Compute edge weights.

Methods Documentation

forward(source, target)

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.