group1_forward_branch¶
- group1_forward_branch(layer, in_tensor, resized_feat)[source]¶
Define group 1 connections.
- Parameters:
layer (torch.nn.Module) – Network layer.
in_tensor (torch.Tensor) – Input tensor.
resized_feat (torch.Tensor) – Resized input.
- Returns:
Output of group 1 layer.
- Return type: