torch_geometric_signed_directed.nn.directed.complex_relu

Classes

complex_relu_layer

The complex ReLU layer from the MagNet: A Neural Network for Directed Graphs. paper.

Module Contents

class complex_relu_layer

Bases: torch.nn.Module

The complex ReLU layer from the MagNet: A Neural Network for Directed Graphs. paper.

complex_relu(real: torch.FloatTensor, img: torch.FloatTensor)

Complex ReLU function.

Arg types:

  • real, imag (PyTorch Float Tensor) - Node features.

Return types:

  • real, imag (PyTorch Float Tensor) - Node features after complex ReLU.

forward(real: torch.FloatTensor, img: torch.FloatTensor)

Making a forward pass of the complex ReLU layer.

Arg types:

  • real, imag (PyTorch Float Tensor) - Node features.

Return types:

  • real, imag (PyTorch Float Tensor) - Node features after complex ReLU.