BVhy.dZddlZddlmZdZdZdZy)z$A library of common shape functions.N) tensor_shapec tttjt|jt|jt j d}g}|D]\}}|j |jl|j!|jdkDr|j|L|j!|jdkDr|j|y|jd|jdk(r|j||jdk(r|j||j|jk(r"|j|j|y|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_longestdimsr Dimensionvalueappend merge_with)shape_xshape_ybroadcasted_dims return_dimsdim_xdim_ys Y/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/framework/common_shapes.py_broadcast_shape_helperrs6   w||$w||$$..q1 345+(nue {{ekk1  U[[1_5! ;; "u{{Q5!4      #))%01 12 cP|j |jyt||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)rrs ris_broadcast_compatiblerGs, ]]gmm3  ' 2$ >>rc|j |jtjSt||}|t d|d|dtj |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. zIncompatible 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 .)rr unknown_shaper ValueError TensorShape)rrrs rbroadcast_shaper!Wsq ]]gmm3  % % '''9+ "#*%y; <<  ! !+ ..r)__doc__r tensorflow.python.frameworkrrrr!rrr%s +4/d ? /r