merge_branch.py

def final_block(pooling_branch_channels=32, pool_type="avg"):
    network = []
    
    #Create subnetwork and add branches
    subnetwork = []
    branch_1 = first_branch()
    branch_2 = second_branch()
    branch_3 = third_branch()
    branch_4 = fourth_branch(pooling_branch_channels=pooling_branch_channels, 
                             pool_type=pool_type)
    subnetwork.append(branch_1)
    subnetwork.append(branch_2)
    subnetwork.append(branch_3)
    subnetwork.append(branch_4)
    
    # Add merging element
    subnetwork.append(gtf.concatenate())
    
    # Add the subnetwork
    network.append(subnetwork)
    return network