o ?ey@s4dZddlZddlmZddZddZdd ZdS) z$A library of common shape functions.N) tensor_shapecCstttjt|jt|jtdd}g}|D][\}}|jdus'|jdurM|jdur7|jdkr7||q|jdurG|jdkrG||q|dq|jdkrX||q|jdkrc||q|j|jkrr|| |qdS|S)aHelper functions for is_broadcast_compatible and broadcast_shape. Args: shape_x: A `TensorShape` shape_y: A `TensorShape` Returns: Returns None if the shapes are not broadcast compatible, a list of the broadcast dimensions otherwise. ) fillvalueN) reversedlist itertools zip_longestdimsrZ DimensionvalueappendZ merge_with)shape_xshape_yZbroadcasted_dims return_dimsZdim_xZdim_yrj/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/framework/common_shapes.py_broadcast_shape_helpers2          rcCs&|jdus |jdur dSt||duS)zReturns True if `shape_x` and `shape_y` are broadcast compatible. Args: shape_x: A `TensorShape` shape_y: A `TensorShape` Returns: True if a shape exists that both `shape_x` and `shape_y` can be broadcasted to. False otherwise. NF)ndimsr)r r rrris_broadcast_compatibleGs rcCsN|jdus |jdurtSt||}|dur"td|d|dt|S)a Returns the broadcasted shape between `shape_x` and `shape_y`. Args: shape_x: A `TensorShape` shape_y: A `TensorShape` Returns: A `TensorShape` representing the broadcasted shape. Raises: ValueError: If the two shapes can not be broadcasted. NzIncompatible shapes for broadcasting. Two shapes are compatible if for each dimension pair they are either equal or one of them is 1. Received: z and .)rrZ unknown_shaper ValueErrorZ TensorShape)r r rrrrbroadcast_shapeWs   r)__doc__rZtensorflow.python.frameworkrrrrrrrrs  2