o to™i£ã@s4ddlmZddlZddlmZGdd„deƒZdS)é)Ú annotationsN)Ú OpRunUnaryc@seZdZddd„ZdS)ÚFlattenNcCsD|p|j}|j}|dkrdn t |d|…¡ t¡df}| |¡fS)Nr)ééÿÿÿÿr)ÚaxisÚshapeÚnpÚprodÚastypeÚintÚreshape)ÚselfÚxrÚirÚ new_shape©rúQ/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_flatten.pyÚ_run s ( z Flatten._run)N)Ú__name__Ú __module__Ú __qualname__rrrrrr sr)Ú __future__rÚnumpyr Úonnx.reference.ops._oprrrrrrÚs