קg 0ddlZddlmZddZddZdZdS)N)reduceTc|r||kr|S||S)N)size contiguousview)tensorrcheck_same_sizes [/var/www/html/ai-engine/env/lib/python3.11/site-packages/torch/autograd/_functions/utils.py maybe_viewr sA6;;==D00      # #D ) ))c|r||kr|S|t|z }dtt ||d|D}t |D]}|dd}|D]}||d}|S)Nc*g|]\}\}}||k|Sr).0dimexpandedoriginals r z"maybe_unexpand..s: %C%(H x     r rF)keepdimT)rrlen enumerateziprangesum)rold_sizer num_unsqueezed expanded_dims_rs r maybe_unexpandr s6;;==H44 ZZ\\CMM1N)2  noo. 9 9* * M> " "..Au--//C.. Mr cfd}d}t|}t|}ttj|}ttj|}||kr d}|dkrd}n2||krd}|dkr|||z d|krd}n||kr d}|dkrd}|st d|d||S)NFTzCNumpy style broadcasting is not supported in ONNX. Input dims are: z, )rroperatormul ValueError)dims1dims2 broadcast supportedlen1len2numel1numel2s r check_onnx_broadcastr-&sII u::D u::D HL% ( (F HL% ( (F d{{ Q;;I  Q;;5/588I E>>I{{!   bRW b b[` b b    r )T)r" functoolsrr rr-rr r r/sa**** 4r