o h@s,ddlmZd ddZd ddZddZd S) )reduceTcCs"|r ||kr |S||SN)size contiguousview)tensorrcheck_same_sizer U/var/www/html/ai/venv/lib/python3.10/site-packages/torch/autograd/_functions/utils.py maybe_viewsr cCs|r ||kr |S|t|}ddtt||d|D}t|D] }|jddd}q(|D] }|j|dd}q4|S)NcSs g|] \}\}}||kr|qSr r ).0dimexpandedoriginalr r r s  z"maybe_unexpand..rF)keepdimT)rr len enumerateziprangesum)rold_sizernum_unsqueezed expanded_dims_r r r r maybe_unexpand s rcCsd}d}t|}t|}tdd|}tdd|}||kr'd}|dkr&d}n#||kr>d}|dkr=|||d|kr=d}n ||krJd}|dkrJd}|sVtd|d||S) NFTcS||Srr xyr r r )z&check_onnx_broadcast..cSrrr rr r r r *r!zCNumpy style broadcasting is not supported in ONNX. Input dims are: z, )rr ValueError)dims1dims2 broadcast supportedlen1len2numel1numel2r r r check_onnx_broadcast$s2r,N)T) functoolsrr rr,r r r r s