scglue.models.nn.GraphAttent.forward

GraphAttent.forward(input, eidx, ewt, esgn)[source]

Forward propagation

Parameters

  • input (Tensor) – Input data (\(n_{vertices} \times n_{features}\))

  • eidx (Tensor) – Vertex indices of edges (\(2 \times n_{edges}\))

  • ewt (Tensor) – Weight of edges (\(n_{edges}\))

  • esgn (Tensor) – Sign of edges (\(n_{edges}\))

Return type

Tensor

Returns

result – Graph attention result (\(n_{vertices} \times n_{features}\))