group2_forward_branch

tiatoolbox . models . architecture . micronet . group2_forward_branch

group2_forward_branch(layer, in_tensor)

Defines group 1 connections.

Parameters

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

• in_tensor (torch.Tensor) – Input tensor.

Returns

Output of group 1 layer.

Return type

torch.Tensor