tfg.image.pyramid.merge

Merges the different levels of the pyramid back to an image.

tfg.image.pyramid.merge(
    levels: List[type_alias.TensorLike],
    name: str = 'pyramid_merge'
) -> tf.Tensor

levels A list containing tensors of shape [B, H_i, W_i, C], where B is the batch size, H_i and W_i are the height and width of the image for the level i, and C the number of channels of the image.
name A name for this op that defaults to "pyramid_merge".

A tensor of shape [B, H, W, C], where B is the batch size, H the height of the image, W the width of the image, and C the number of channels of the image.

ValueError If the shape of the elements of levels is not supported.