o ei@sdZddlZddlZddlmZddZGdddeeZGd d d e e Z d d Z d dZ d'ddZd(ddZddZddZddZddddZddZdd Zd!d"Zd)d#d$Zd%d&ZdS)*zKAssorted utilities, which do not need anything other then torch and stdlib.N) _dtypes_implcCs2t|trdSzt|WdStyYdSw)NFT) isinstancestrlen Exception)seqr \/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/_numpy/_util.py is_sequence s   r c@ eZdZdS) AxisErrorN__name__ __module__ __qualname__r r r r r r c@r )UFuncTypeErrorNrr r r r rrrcCs |dur|j|kr||}|SN)dtypeto)tensorrr r r cast_if_neededs rcCs$t|jdkr|tj}|S)N)r _categoryrrdefault_dtypes float_dtype)xr r r cast_int_to_float&srcCs@| |kr |ksntd|d||dkr||7}|S)Nzaxis z) is out of bounds for array of dimension r)r )axndimargnamer r r normalize_axis_index.s r"Fcst|ttfvrzt|g}Wn tyYnwtfdd|D}|sCtttt |t|krCr?t ddt d|S)a Normalizes an axis argument into a tuple of non-negative integer axes. This handles shorthands such as ``1`` and converts them to ``(1,)``, as well as performing the handling of negative indices covered by `normalize_axis_index`. By default, this forbids axes from being specified multiple times. Used internally by multi-axis-checking logic. Parameters ---------- axis : int, iterable of int The un-normalized index or indices of the axis. ndim : int The number of dimensions of the array that `axis` should be normalized against. argname : str, optional A prefix to put before the error message, typically the name of the argument. allow_duplicate : bool, optional If False, the default, disallow an axis from being specified twice. Returns ------- normalized_axes : tuple of int The normalized axis index, such that `0 <= normalized_axis < ndim` c3|] }t|VqdSr)r".0rr!r r r [z'normalize_axis_tuple..zrepeated axis in `z ` argumentz repeated axis) typetuplelistoperatorindex TypeErrorrsetmapint ValueError)axisr r!allow_duplicater r&r normalize_axis_tuple7s r5cCs(|dur|St|dkrtd|dS)Nrzdoes not handle tuple axisr)rNotImplementedErrorr3r r r allow_only_single_axisds  r8csTtttfvr ftt|}t|t|fddt|D}|S)Ncs g|] }|vr dntqSr)nextr$r3shape_itr r ss z expand_shape..)r)r+r*rr5iterrange) arr_shaper3out_ndimshaper r;r expand_shapels rCcCs<|durd|}||}|St|j|}||}|S)Nr9)expand contiguousrCrBreshape)rr3r rBr r r apply_keepdimsws rGr7cGs*|durtdd|D}|dfS||fS)z#Flatten the arrays if axis is None.Ncss|]}|VqdSr)flatten)r%arr r r r'sz$axis_none_flatten..rr*)r3tensorsr r r axis_none_flattensrLcCs>tj}||j||dstd|jd|d|dt||S)aDtype-cast tensor to target_dtype. Parameters ---------- t : torch.Tensor The tensor to cast target_dtype : torch dtype object The array dtype to cast all tensors to casting : str The casting mode, see `np.can_cast` Returns ------- `torch.Tensor` of the `target_dtype` dtype Raises ------ ValueError if the argument cannot be cast according to the `casting` rule )castingzCannot cast array data from z to z according to the rule '')r can_cast_implrr.r)t target_dtyperMcan_castr r r typecast_tensors  rScstfdd|DS)Nc3r#r)rS)r%rPrMrQr r r'r(z#typecast_tensors..rJ)rKrQrMr rTr typecast_tensorssrUc CsHzt|}W|Sty#}zd|dt|d}t|d}~ww)Nzfailed to convert z! to ndarray. Internal error is: .)torch as_tensorrrr6)objremesgr r r _try_convert_to_tensors r\c Cst|tjr |}nt}tttjz t|}Wt|nt|wt ||}||j }|dkr@| d||j }zt |}Wn tyQd}Ynw|rX|}|S)aThe core logic of the array(...) function. Parameters ---------- obj : tensor_like The thing to coerce dtype : torch.dtype object or None Coerce to this torch dtype copy : bool Copy or not ndmin : int The results as least this many dimensions is_weak : bool Whether obj is a weakly typed python scalar. Returns ------- tensor : torch.Tensor a tensor object with requested dtype, ndim and copy semantics. Notes ----- This is almost a "tensor_like" coercive function. Does not handle wrapper ndarrays (those should be handled in the ndarray-aware layer prior to invoking this function). rr9F)rrWTensorget_default_dtypeset_default_dtyperget_default_dtype_forfloat32r\rr viewrBboolr2clone)rYrcopyndminr default_dtype ndim_extrar r r _coerce_to_tensors&      ricGsddlm}t|dkrtSt|dkr<|d}t||r!|jSt|tr:g}|D] }t|}||q*t|S|St|tsKt dt |j t|S)zHConvert all ndarrays from `inputs` to tensors. (other things are intact)r)ndarrayrz#Expected inputs to be a tuple, got ) _ndarrayrjrr2rrr*ndarrays_to_tensorsappendAssertionErrorr)r)inputsrjinput_result sub_input sub_resultr r r rls&       rlr)NF)NFr)__doc__r,rWrr r2 IndexErrorr r. RuntimeErrorrrrr"r5r8rCrGrLrSrUr\rirlr r r r s(    -     >