morphoclass.layers.cat module

Implementation of the Cat layer.

class morphoclass.layers.cat.Cat

Bases: torch.nn.modules.module.Module

Concatenate the outputs of multiple parallel layers.

forward(x_list)

Perform the forward pass.

Parameters

x_list (iterable of torch.Tensor) – An iterable of outputs of a number of torch layers.

Returns

The concatenated input tensors.

Return type

torch.Tensor

training: bool