group3_forward_branch

tiatoolbox . models . architecture . micronet . group3_forward_branch

group3_forward_branch(layer, main_feat, skip)

Defines group 1 connections.

Parameters

• layer (torch.nn.Module) – Network layer.

• main_feat (torch.Tensor) – Input tensor.

• skip (torch.Tensor) – Skip connection.

Returns

Output of group 1 layer.

Return type

torch.Tensor