o Zh@s4ddlZddlmZd ddZd ddZdd ZdS) N)reduceTcCs"|r ||kr |S||S)N)size contiguousview)tensorrcheck_same_sizerN/var/www/auris/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|qSrr).0dimexpandedoriginalrrr s  z"maybe_unexpand..rF)ZkeepdimT)rr len enumerateziprangesum)rZold_sizerZnum_unsqueezedZ expanded_dims_r rrr maybe_unexpand s rcCsd}d}t|}t|}ttj|}||krd}|dkrd}n#||kr6d}|dkr5|||d|kr5d}n ||krBd}|dkrBd}|sNtd|d||S)NFTzCNumpy style broadcasting is not supported in ONNX. Input dims are: z, )rroperatormul ValueError)Zdims1Zdims2 broadcast supportedZlen1Zlen2Znumel2rrr check_onnx_broadcast&s0 r)T)r functoolsrr rrrrrr s