U d@s,ddlmZd ddZd ddZddZd S) )reduceTcCs"|r||kr|S||SN)size contiguousview)tensorrcheck_same_sizer C/tmp/pip-unpacked-wheel-ua33x9lu/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|||S)NFTcSs||Srr xyr r r %z&check_onnx_broadcast..cSs||Srr rr r r r&rzINumpy style broadcasting is not supported in ONNX. Input dims are: {}, {})rr ValueErrorformat)Zdims1Zdims2 broadcast supportedZlen1Zlen2Znumel1Znumel2r r r check_onnx_broadcast s0r"N)T)T) functoolsrr rr"r r r r s