2j<BddlmZddlZddlmZGddeZdS)) annotationsN)OpRunceZdZ ddZdS) MaxUnpoolNcvt|jdz }|dkrtd|d|p|j}|p|j}|p|j}| d|D}|)dt t|dzD}tjt|jftj }|jd|d<|jd |d <t t|D]O} |j| dzd z || z|| |t|| zzz || z|| dz<P||} n|} tj |j} tj tj |f|j } | } | }t | D]}||| | |<| t|} tj | | j }td |D}| ||<|fS) NzUnsupported pooling size z for operator MaxUnpool.cg|]}dS).0ds QC:\Projects\Historical\venv\Lib\site-packages\onnx/reference/ops/op_max_unpool.py z"MaxUnpool._run..s///Qq///cg|]}dS)rr r s rrz"MaxUnpool._run..s<<z!MaxUnpool._run..8s*;;quQ{{;;;;;;r)lenshapeNotImplementedError kernel_shapepadsstridesrangenpemptyint64prodzerosrflattenreshapetuple)selfXindices output_shaperrr pooling_dimsinferred_shapedimrtotal_elementsYI_dataX_datacur_elemresslicess r_runzMaxUnpool._run sV17||a' !  %RLRRR $8t'8  ty)T\ ?//,///G <<s#$ 37 # #  "EE E)) Hbgn--/qw ? ? ?""n-- 3 3H"("2AfX   IIeN++ , ,huAG,,,;;N;;;;;F v r)NNNN)__name__ __module__ __qualname__r6r rrrr s+SW......rr) __future__rnumpyr onnx.reference.op_runrrr rrr=sk#"""""''''''//////////r