outBlock

class colibri.models.custom_layers.outBlock(in_channels, out_channels, activation=None)

Bases: Module

Convolutional Block with 1x1 kernel and without activation

Parameters:

• in_channels (int) – number of input channels

• out_channels (int) – number of output channels

• activation (str, optional) – activation function. Defaults to None.

forward(x)

Forward pass of the outBlock.

Parameters:

x (torch.Tensor) – Input tensor

Returns:

Output tensor

Return type:

torch.Tensor