def conv_ch_coupling(input_, dim, name, use_batch_norm=True, train=True, weight_norm=True, reverse=False, residual_blocks=5, bottleneck=False, use_aff=True, change_bottom=True, skip=True):
 if use_aff:
  return conv_ch_aff_coupling(input_=input_, dim=dim, name=name, use_batch_norm=use_batch_norm, train=train, weight_norm=weight_norm, reverse=reverse, residual_blocks=residual_blocks, bottleneck=bottleneck, change_bottom=change_bottom, skip=skip)
 else:
  return conv_ch_add_coupling(input_=input_, dim=dim, name=name, use_batch_norm=use_batch_norm, train=train, weight_norm=weight_norm, reverse=reverse, residual_blocks=residual_blocks, bottleneck=bottleneck, change_bottom=change_bottom, skip=skip)

def conv_ch_coupling ( input_ , dim , name , use_batch_norm = True , train = True , weight_norm = True , reverse = False , residual_blocks = 5 , bottleneck = False , use_aff = True , change_bottom = True , skip = True ) : if use_aff : return conv_ch_aff_coupling ( input_ = input_ , dim = dim , name = name , use_batch_norm = use_batch_norm , train = train , weight_norm = weight_norm , reverse = reverse , residual_blocks = residual_blocks , bottleneck = bottleneck , change_bottom = change_bottom , skip = skip ) else : return conv_ch_add_coupling ( input_ = input_ , dim = dim , name = name , use_batch_norm = use_batch_norm , train = train , weight_norm = weight_norm , reverse = reverse , residual_blocks = residual_blocks , bottleneck = bottleneck , change_bottom = change_bottom , skip = skip )
coupling with channel-wise splitting .