morphoclass.layers.bidirectional_res_block module

Implementation of the BidirectionalResBlock layer.

class morphoclass.layers.bidirectional_res_block.BidirectionalResBlock(c_in, c_out)

Bases: torch.nn.modules.module.Module

A BidirectionalBlock layer with a skip connection.

Parameters

  • c_in (int) – The number of input channels.

  • c_out (int) – The number of output channels.

forward(x, edge_index)

Perform the forward pass.

Parameters

  • x (torch.Tensor) – A batch of node features.

  • edge_index (torch.Tensor) – A batch of adjacency matrices.

Returns

x – The output node features.

Return type

torch.Tensor

training: bool