baby.layers.unet_block

baby.layers.unet_block(input_tensor, layer_sizes, **hyperparams)