o b¦µi<ã@sÈddlZddlmZddlmZddlmZddlmZddlmZd&dd „Z d&d d „Z d d „Z dd„Z dd„Z dd„Zdd„Zd'dd„Zd(dd„Zdd„Zd&dd„Zd)d d!„Zd*d"d#„Zd&d$d%„ZdS)+éN)Úconfig)Ústandardize_dtype)Údtypes)Úcast)Úconvert_to_tensorFcCstjj||dS©N©Úupper)ÚtorchÚlinalgÚcholesky©Úxr ©rúR/home/ubuntu/.local/lib/python3.10/site-packages/keras/src/backend/torch/linalg.pyr sr cCstj||dSr)r Úcholesky_inverser rrrrórcCs t |¡S©N)r Údet©rrrrrs rcCó tj |¡Sr)r r Úeigrrrrró rcCrr)r r ÚeighrrrrrrrcCrr)r r ÚinvrrrrrrrcCstj |¡\}}||dfS)Né)r r Ú lu_factor)rÚLUÚpivotsrrrr"s rcCsLt|ƒ}t|jƒdkrt ¡}nt |jt¡}t||ƒ}t j j ||||dS)NÚint64)ÚordÚdimÚkeepdim) rrÚdtyperÚfloatxrÚ result_typeÚfloatrr r Únorm)rr ÚaxisÚkeepdimsr#rrrr'(s   r'ÚreducedcCs&|dvr td|›ƒ‚tjj||dS)N>r*Úcompletez]`mode` argument value not supported. Expected one of {'reduced', 'complete'}. Received: mode=)Úmode)Ú ValueErrorr r Úqr)rr,rrrr.2sþÿr.cCstj ||¡Sr)r r Úsolve)ÚaÚbrrrr/<rr/cCsN|j|jdkrtj|dd}tjj||| djddStjj||| dS)Nréÿÿÿÿ)r(r)Úndimr Ú unsqueezer Úsolve_triangularÚsqueeze)r0r1Úlowerrrrr5@s ÿr5TcCs |stj |¡Stjj||dS)N)Ú full_matrices)r r ÚsvdvalsÚsvd)rr8Ú compute_uvrrrr:Is r:cCs&t|ƒ}t|ƒ}tjj|||ddS)N)Úrcondr)rr r Úlstsq)r0r1r<rrrr=Osr=cCstjj||||dS)N)Úhas_aux)r ÚfuncÚjvp)ÚfunÚprimalsÚtangentsr>rrrr@Usr@)F)NNF)r*)TTr)r Úkeras.src.backendrrÚkeras.src.backend.commonrÚkeras.src.backend.torch.corerrr rrrrrrr'r.r/r5r:r=r@rrrrÚs(