upBlock

class colibri.models.custom_layers.upBlock(in_channels)

Bases: Module

Spatial upsampling and then convBlock

Parameters:

in_channels (int) – number of input channels

forward(x1, x2)

Forward pass of the upBlock.

Parameters:

• x1 (torch.Tensor) – Input tensor

• x2 (torch.Tensor) – Input tensor

Returns:

Output tensor

Return type:

torch.Tensor