
    h                     \    S SK r S SKr\ R                  " \5      rS rS rS rS r	S r
S rg)    Nc                     [         R                  " U 5      n[         R                  " U5      (       d  [         R                  " U5      (       a  [        R                  S5        U $ )Nz!NaN or Inf found in input tensor.)npsumisnanisinfloggerwarning)arraytmps     L/var/www/fran/franai/venv/lib/python3.13/site-packages/tensorboardX/x2num.py	check_nanr   
   s:    
&&-C	xx}}:;L    c                 H   [        U [        5      (       a  [        [        R                  " U 5      5      $ [        U [        R
                  5      (       a  [        U 5      $ [        R                  " U 5      (       a   [        [        R                  " U /5      5      $ S[        [        U 5      5      ;   a  [        [        U 5      5      $ S[        [        U 5      5      ;   a  [        [        U 5      5      $ S[        [        U 5      5      ;   a  [        [        U 5      5      $ S[        [        U 5      5      ;   a  [        [        R                  " U 5      5      $ S[        [        U 5      5      ;   a  [        [        R                  " U 5      5      $ [        S[        U 5       S35      e)NtorchchainermxnetjaxpaddlezGot z+, but expected numpy array or torch tensor.)
isinstancelistr   r   r
   ndarrayisscalarstrtypeprepare_pytorchprepare_chainerprepare_mxnetNotImplementedErrorxs    r   make_npr!      s%   !T!%%!RZZ  |	{{1~~1#''#d1g,+,,CQL +,,#d1g,q)**DG!%%3tAw<!%%

tAwiBCE Er   c                     SS K n[        XR                  R                  5      (       a  U R                  n U R                  5       R                  5       n U $ Nr   )r   r   autogradVariabledatacpunumpy)r    r   s     r   r   r   &   s9    !^^,,--FF	AHr   c                     SS K ng r#   )theano)r    r*   s     r   prepare_theanor+   .   s    r   c                 &    U R                  5       n U $ )N)asnumpyr   s    r   r   r   3   s    			AHr   c                 X    SS K nUR                  R                  U R                  5      n U $ r#   )r   cudato_cpur&   )r    r   s     r   r   r   8   s"    AFF#AHr   )loggingr(   r   	getLogger__name__r   r   r!   r   r+   r   r    r   r   <module>r5      s<     			8	$E*	

r   