U ʹj@s,ddlmZd ddZd ddZddZd S) )reduceTcCs"|r||kr|S||SN)size contiguousview)tensorrcheck_same_sizer ^/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/autograd/_functions/utils.py maybe_viewsr cCs|r||kr|S|t|}ddtt||d|D}t|D]}|jddd}qP|D]}|j|dd}qh|S)NcSs g|]\}\}}||kr|qSr r ).0dimexpandedoriginalr r r s z"maybe_unexpand..rF)ZkeepdimT)rr len enumerateziprangesum)rZold_sizerZnum_unsqueezedZ expanded_dims_r r r r maybe_unexpand s rcCsd}d}t|}t|}tdd|}tdd|}||krNd}|dkrd}nF||kr|d}|dkr|||d|krd}n||krd}|dkrd}|std|d||S) NFTcSs||Srr xyr r r )z&check_onnx_broadcast..cSs||Srr rr r r r*rzCNumpy style broadcasting is not supported in ONNX. Input dims are: z, )rr ValueError)Zdims1Zdims2 broadcast supportedZlen1Zlen2Znumel1Znumel2r r r check_onnx_broadcast$s.r!N)T)T) functoolsrr rr!r r r r s