|/hiddlZddZy)Nctj|}|%tdt|jD}|t |z}|tj tj|||z }tj|}tj tj|||}||z }t|jdk(r|j}|S)a Compute the softmax of each element along an axis of X. Parameters ---------- X: ND-Array. Probably should be floats. theta (optional): float parameter, used as a multiplier prior to exponentiation. Default = 1.0 axis (optional): axis to compute values along. Default is the first non-singleton axis. Returns an array the same size as X. The result will sum to 1 along the specified axis. c38K|]}|ddkDs |dyw)rN).0js D/server/volume/plugins/@scrypted/onnx/zip/unzipped/common/softmax.py zsoftmax..sAQ!qAaDAs  )axisr) np atleast_2dnext enumerateshapefloat expand_dimsmaxexpsumlenflatten)Xthetar yax_sumps r softmaxrs" aA |A)AGG"4AA E%LA BNN266!D14 88A q A^^BFF1T2D 9F F A 177|qaiik! H)g?N)numpyr rrrr r s ) r