o "ÕjC ã@sHddlZddlZddlmZmZddlmZdd„Zdd„Z dd „Z dS) éN)Ú check_dtypeÚ convert_dtype)ÚVariablecCs’|dur dg}|St|ttjtjfƒr|g}|St|ttjjfƒrBd|_ t |j dddgdd|d ƒt |j ƒdkr@t  |d¡}|St|ƒg}|S) z] Convert out_size(int, np.int32, np.int64, Variable) to list in imperative mode. NrTÚout_sizeÚint32Úint64Úop_typeú(When type of out_size inú is Variable.))Ú isinstanceÚintÚnprrrÚpaddleZpirZOpResultÚ stop_gradientrÚdtyperÚcast)rr©rúg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/geometric/message_passing/utils.pyÚconvert_out_size_to_lists(ñó û  ÿrcCs”|dur dg|d<dSt|ttjtjfƒr|g|d<dSt|tƒrFd|_t|jdddgdd|d ƒt |jƒdkr@t   |d¡}||d <dSt d ƒ‚) zk Convert out_size(int, np.int32, np.int64, Variable) to inputs and attrs in static graph mode. NrrTrrrr r ZOut_sizez'Out_size only supports Variable or int.) r r r rrrrrrrrrÚ TypeError)ZinputsÚattrsrrrrrÚget_out_size_tensor_inputs/s"  û  rc Csþt|jƒdkrt |ddg¡}t|jƒdkrt |ddg¡}t |¡}t |¡}t|jƒt|jƒkr{tt|jƒt|jƒƒ}|t|jƒ}|t|jƒ}|dgdg|t|dd…ƒ}|dgdg|t|dd…ƒ}t ||¡}t ||¡}||fS)zm Expand dims to ensure there will be no broadcasting issues with different number of dimensions. ééÿÿÿÿrN)ÚlenÚshaperZreshapeÚmaxÚlist) ÚxÚyZx_shapeZy_shapeZ max_ndimsZ x_pad_ndimsZ y_pad_ndimsZ new_x_shapeZ new_y_shaperrrÚreshape_lhs_rhsHsB  ÿÿýýùÿ ÿÿýýùÿ r ) Únumpyr rZpaddle.base.data_feederrrZpaddle.base.frameworkrrrr rrrrÚs