U yh @s4ddlZddlmZd ddZd ddZdd ZdS) N)reduceTcCs"|r||kr|S||S)N)size contiguousview)tensorrcheck_same_sizerQ/var/www/html/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|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|}ttj|}||krJd}|dkrd}nF||krxd}|dkr|||d|krd}n||krd}|dkrd}|std|d||S)NFTzCNumpy style broadcasting is not supported in ONNX. Input dims are: z, )rroperatormul ValueError)Zdims1Zdims2 broadcast supportedZlen1Zlen2Znumel1Znumel2rrr check_onnx_broadcast&s.  r)T)T)r functoolsrr rrrrrr s