3j}kSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKJrJ r J r SSK J r SSK JrJrJrJr SSKJrJrJrJrJr SSKrSSKJr SSKJr SSKJs J r! SSKJ"r"J#r#J$r$J%r% SSKJ&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.J/r/J0r0J1r1J2r2J3r3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;Jr>J?r?J@r@JArAJBrBJCrC SS KDJErE /S QrF\RrG\RRrI\RRRrLS rMS rNS rOSS.SjrPSSKQJRrR \S"5rT\TSSS.S\4SjjrUSrVSrW\U"\,RSS9S5rY\U"\,R5S5r[\U"\,R5S5r\\U"\,R5S5r]\U"\,R5S5r^\U"\,R5S5r_\U"\,R5S5r`\U"\,R5S 5rb\U"\,RSS9S!5rc\R"\LR5S"\:4S#j5rd\R"\LR5\C"5S"\:4S$j55re\U"\,R5S%5rf\U"\,R5S&5rg\U"\,R5S'5rh\U"\,R5S(5ri\U"\,R5S)5rj\U"\,R5S*5rk\U"\,R5S+5rl\U"\,R5S,5rm\U"\,R5S-5rn\C"5\A"S.\,RS/9S0\:S1\4S\:4S2j55rpS0\:S1\4S\:4S3jrq\R"\LR5\C"5S"\:S\:4S4j55rr\U"\,RSS9S55rs\U"\,RSS9S6\:S\:4S7j5rtS0\:S\:4S8jru\U"\,RSS99S0\:S\:4S:j5rw\U"\,R5S0\:S\:4S;j5rx\U"\,RSS9S0\:S\:4S<j5ry\U"\,RSS9S0\:S\:4S=j5rz\U"\,R5S0\:S\:4S>j5r{\V"\RRS?5r~\U"\,RSS99S0\:S\:4S@j5r\U"\,R\LGRS99SA5r\U"\,R5SB5r\U"\,R5SC5r\U"\,R5SD5r\U"\,R5SE5r\U"\,R5SF5r\C"5GSS0\:SG\SH\GRS-S\:4SIjj5r\R"\LGR5\C"5\A"SJ\,RS/9GSSK\:SG\*SL\S\:4SMjj555r\R"\LGR5\C"5GSS0\:SN\4S-SO\4S-SP\4S-S\:4 SQjj55rS0\:4SRjr\U"\,R\SS9ST5rS0\:S\:4SUjrS0\:S\:4SVjr\U"\,R5SW5r\R"\LGR"5\C"5\A"S.\,RS/9SSX.S0\:SY\S\:4SZjj555r\U"\,R5S[5r\U"\,R5S0\:S\:4S\j5r\U"\,RSS9S]5r\U"\,RSS9S^5r\U"\,RSS9S_5r\U"\,R5S`5r\U"\,R5Sa5r\U"\,R5Sb5r\U"\,R5Sc5r\U"\,GR6SS99S0\:S\:4Sdj5r\U"\,R5Se5r\U"\,R5Sf5r\U"\,RSS9Sg5rSK\:S\:4Shjr\TSSSSSS4S\4SijjrS0\:\4-Sj\:\4-S\GR4Skjr\R"\LGRF5\C"5\A"Sl\,RS/9SSm.S0\:\4-Sj\:\4-Sn\4S-4Sojj555r\"\,RSSSp9Sq5r\"\,RSr9S0\:Sj\:S\:4Ssj5r\"\,RSr9S0\:Sj\:S\:4Stj5r\"\,RSr9S0\:Sj\:S\:4Suj5r\"\,RSr9S0\:Sj\:S\:4Svj5r\"\,RSr9S0\:Sj\:S\:4Swj5r\"\,RSSx9S0\:\4-Sj\:\4-4Syj5r\R"\LGRV5\C"5SSz.S0\:\4-Sj\:\4-S{\S-4S|jj55r\"\,RSSx9S0\:Sj\:S\:4S}j5r\"\,GR6Sr9S0\:\4-Sj\:\4-S\:4S~j5r\C"5S0\:\4-Sj\:\4-S\G4Sj5r\"\RXRSSS9S0\:\4-Sj\:\4-4Sj5rS0\GSj\GS\G4SjrS0\GSj\GS\G4Sjr\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\R"\LGRl5\C"SS5SK\:S\\:\:44Sj55r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSSx9S0\:Sj\:S\:4Sj5r\"\,RSSx9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S"\:S\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5rS\S0\:Sj\:S\S\SS4 SjrGSS0\:Sj\:S\S\S\S\:4 Sjjr\"\,RSSSp9S0\:Sj\:4Sj5r\"\,RSSx9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSr9S0\:Sj\:4Sj5r\U"\,R5S0\:4Sj5r\"\,RSr9S0\:Sj\:4Sj5r\"\,RSr9S0\:Sj\:4Sj5r\"\,RSSx9S0\:Sj\:S\:4Sj5r\"\,RSr9S0\:Sj\:S\:4Sj5r\"\,RSr9S0\:Sj\:S\:4Sj5r\"\,RSS9S0\:Sj\:S\:4Sj5r\"\,RSSx9S0\:Sj\:S\:4Sj5r\"\,RSSSp9S0\:Sj\:S\:4Sj5r\"\,RSr9S0\:Sj\:S\:4Sj5r\R"\LGR5\C"5GSS0\:\4-Sj\:\4-Sn\44Sjj55r\R"\LGR5\C"5\A"Sl\,RS/9SSm.S0\:\4-Sj\:\4-Sn\44Sjj555r\"\,RSSSS9S0\:Sj\:S\:4Sj5r\R"\LGR5\C"5\A"Sl\,RS/9S0\:\4-Sj\:\4-4Sj555r\"\RXRSSS9S0\:\4-Sj\:\4-4Sj5r\R"\LGR5\C"5\A"S\,RS/9SS.SK\:S\:S\:S1\4S\:4 Sjj555r\R"\LGR5\C"5\A"S\,RS/9SS.SK\:S\:S\:S1\4S\:4 Sjj555r\R"\LGR5\C"5\A"S\,RS/9GSS0\:S\;S-S\;S-S\:4Sjj555r\R"\LGR5\C"5GSSK\:S\;S-S\:4Sjj55r\R"\LGR5\C"5GSSK\:S\;S-S\:4Sjj55r\R"\LGRGR5S\GS\\GS44Sj5r\R"\LGRGR5\R"\LGRGR5\R"\LGRGR5\R"\LGRGR5\R"\LGRGR5\C"SS9\A"Sl\,RS/9GSS\GS0\;S-Sj\;S-4Sjj5555555r\R"\LGR5\C"5\GRS.S0\:S\GRS\:4Sjj55rSS.S0\GSj\G4Sjjr\R"\LGR5S0\:S\44Sj5rGSS0\:S\'S-SH\GRS-S\S-S\GRS-S\GRS-S\S-S\S\4Sjjr\S5r\GRGSS\GRSH\GRS\S\S\GRS-S\\\44 Sjj5r\GRGSS\SH\GRS\S\S\GRS-S\\\44 Sjj5r\GRGSSH\GRS\S\S\GRS-S\\\44 Sjj5r\GRGSS\GS\S\S\GRS-S\\\44 Sjj5rS0\GS\4SjrS0\:S\:4SjrSSSSSSS.S0\:S\S\S\S\*S-S\SH\GRS-S\GS-S\6S\:4SjjrGSSjr\R"\LGR5\C"5GSS0\:SG\*S-SL\S\:4Sjj55r\R"\LGR5\C"5GSS0\:SG\*S-SL\S\:4Sjj55r\R"\LGRGR\LGRGR/5GSSSS.S0\:SG\S-\\-SL\SH\GRS-S\GS-S\:4 Sjjj5rS0\GS\G4Sjr\R"\LGR5GSSSS.S0\:SG\S-\\-SL\S\GS-S\:4 Sjjj5r\R"\LGR5GSSS.S0\:SG\*S-SL\S\GS-S\:4 Sjjj5r\R"\LGR5GSSS.S0\:SG\*S-SL\S\GS-S\:4 Sjjj5rGSSjGr\R"\LGR5\C"5GSSS.S0\:SG\*S-S\S-SL\S\4S-S\:4 Sjjj55Gr\R"\LGR5\C"5GSSS.S0\:SG\S-\\-S\S-SL\S\4S-S\:4 Sjjj55Gr\R"\LGR5GSSSS.S0\:SG\*S-SL\S\:4Sjjj5Gr\R"\LGR5\C"SS5GSSSSS.S0\:SG\*S-S\S-SL\S\4S-4 Sjjj55Gr\R"\LGR 5\C"SS5GSSS.S0\:SG\*S-S\S-SL\S\4S-4 Sjjj55Gr\R"\LGR 5\C"5\A"S\,RS/9SSS.SK\:S\:S\:S\4Sn\4S\:4 Sjj555GrS\:\ \:-S\:S\:\\:S4-4SjGrS\SG\S\:S\:4SjGrS\:\ \:-S\:S\:\\:S4-4GSjGr S\:\ \:-S\:S\:\\:S4-4GSjGr GSS0\:GS\7GS\8GS\S-S\:4 GSjjGr \R"\LGR5\C"5GSS"\:GS\:GS\7GS\8GS\S-S\:4 GSjj55Gr S\74GSjGr \LGRGRGR\IGR 5\LGRGRGR\IGR"5S\\:4GS j55GrS0\:GS\7S\:4GS jGr\R"\LGR&5\C"5\A"GS \,RS/9GSGS \\S-GS?\S-GS@\S-S\G4GSAjj5Gr,\LGR\GRGR\IGR 5GSS"\GGS7\GS8\S-GS9\S-GS:\GS-GS;\GS=\GS>\S-GS!\S-S\G4GSBjj5Gr.GSC\7GSD\8GSE\GS\GSF\4 GSGjGr/\R"\LGR`5\C"5S0\GS\G4GSHj55Gr0S0\:GSI\S\:4GSJjGr1S0\:GSI\S\:4GSKjGr2S0\:GSL\7S\:4GSMjGr3SK\:S\:S\:4GSNjGr4\R"\LGRj5\C"5GSS0\:GSO\*S\*S\:4GSPjj55Gr5\R"\LGRl5\C"5GSS0\:GSR\S\)S\:4GSSjj55Gr6GS \S6\9SG\GS]\9GS^\94GS`jGr?\R"\LGR5\C"5S6\9SG\GS]\9S1\4\9-4GSaj55Gr@\R"\LGR5S6\9SG\GS]\9S1\4\9-4GSbj5GrAS6\9SG\GS]\9S1\4\9-GSc\4 GSdjGrB\C"5SSm.S6\9SG\GS]\9GS^\9Sn\44 GSejj5GrC\R"\LGR5\C"5S6\9SG\GS]\94GSfj55GrD\R"\LGRGR5GSS0\:SG\*S-S\:4GSgjj5GrE\R"\LGR5GSSK\GGSh\\SG\S\\G4GSijj5GrGGSS0\:GSj\G\*-SG\S\\:S44GSkjjGrHS0\:GSj\*S\\:S44GSljGrIS0\:GSj\*S\\:S44GSmjGrJ\R"\LGRGR5\C"5GSSK\:GSn\S\:4GSojj55GrK\R"\LGR5\C"5GSS"\:GS\:GSn\GSp\GSq\S\:4 GSrjj55GrM\R"\LGR5GSSK\:GSn\GSp\GSq\S\:4 GSsjj5GrN\R"\LGR5\C"5GSGS[\:GSn\GSp\GSq\S\:4 GStjj55GrO\R"\LGR5\C"5GS \\:S\:4GSuj55GrQGS \\:S\:4GSvjGrPS0\:GSw\*S\<4GSxjGrR\R"\LGRGR5S0\:4GSyj5GrSS0\:S\:4GSzjGrT\R"\LGR5S0\:S\:4GS{j5GrU\R"\LGR5S0\:GS|\GSp\S\:4GS}j5GrVG\VGrW\R"\LGR5SK\:GSE\GS\GSF\S\:4 GS~j5GrX\R"\LGR5\C"5SK\:GSE\GS\GSF\4GSj55GrYSSS.S0\:SG\SH\GRS-S\GS-S\:4 GSjjGrZ\R"\LGR5SSS.S0\:SG\SH\GRS-S\GS-S\:4 GSjj5Gr[\R"\LGR5SSS.S0\:SG\SH\GRS-S\GS-S\:4 GSjj5Gr\\R"\LGR5S0\:SG\S\:4GSj5Gr]\R"\LGRGR5S0\:GSL\7\\7-S\:4GSj5Gr^SK\:S\:S\:4GSjGr_S0\:S\:4GSjGr`\C"5GSS0\RGS\RSG\S-S\R4GSjj5Gra\C"5S\GRSSS\GR.GS.SH\GRS-S\GRS\'S-GS\S\S\GRS\:4GSjj5Grc\C"5S\GRSSS4SH\GRS-S\GRS\'S-GS\S\S\:4 GSjj5Grd\R"\LGR5\C"5SSSSGS.S0\:GS\7SH\GRS-S\GRS-S\'S-S\S\:4GSjj55Gre\R"\LGR5\C"5SSSSGS.S0\:GS\7GS\8SH\GRS-S\GRS-S\'S-S\S\:4GSjj55Grf\R"\LGRGR5\C"5S\GRSSSGS.SH\GRS-S\GRS\'S-S\GS\S\:4 GSjj55Grg\R"\LGR5\C"5SSSSSGS.S0\:GS\7SH\GRS-S\GRS-S\'S-S\GS\S\:4GSjj55Grh\R"\LGRGR5\C"5S\GRSSSGS.SH\GRS-S\GRS\'S-S\GS\S\:4 GSjj55Gri\R"\LGR5\C"5SSSSSGS.S0\:GS\7SH\GRS-S\GRS-S\'S-S\GS\S\:4GSjj55Grj\R"\LGR5\C"5SSSSGS.S0\:GS\7GS\4SH\GRS-S\GRS-S\'S-S\S\:4GSjj55Grk\LGRGRGR\IGR 5GSGS\:S\:S\GRS-S\:4GSjj5Grl\R"\LGR5\C"5SSSSS\GRGS.S0\:SH\GRS-S\'S-S\GRS-S\GS\S\GRS\:4GSjj55Grm\R"\LGRGR\LGRGR/5\C"5GSS\GRSSSGS.GS \4GS\4S-GSF\4SH\GRS-S\GRS\'S-S\GS\S\:4GSjjj55Grn\R"\LGR5\C"5\A"GS\,RS/9GS \GGS\GGS(\G\4-4GSj555Grq\R"\LGR5\C"5SS\GRSSGS.GS \4\:-GS\4\:-GS\4SH\GRS-S\'S-S\GRS\GS\S\:4GSjj55Grr\R"\LGR5\C"5GSSS\GRSSGS.GS \4\:-GS\4\:-GS\4GS\4SH\GRS-S\'S-S\GRS\GS\S\:4GSjjj55Grs\GS \ \:GS\4GSj5Grt\GS \:GS\4GSj5Grt\R"\LGR5GS \:\\:-\\:-GS\S\\:4GSj5GrtS"\:GS\\)-GS\\)-S\:4GSjGru\R"\LGR5\C"5SS\GRSSGS.GSL\7\\7-GS\8SH\GRS-S\'S-S\GRGS\S\S\:4GSjj55GrvGS\S\GR4GSjGrw\R"\LGR5\C"5GSS\GRSSSGS.GS\GS\S-SH\GRS-S\GRS\'S-S\GS\S\:4GSjjj55Grx\R"\LGRGR\LGRGR/5\C"5S\GRSSSGS.GSL\7GS\4SH\GRS-S\GRS\'S-S\GS\S\:4GSjj55GrySSSSS\GRGS.S0\:GS\4SH\GRS-S\GRS-S\'S-S\GS\S\GRS\:4GSjjGrz\R"\LGR5\C"5SSSSS\GRGS.S0\:SH\GRS-S\GRS-S\'S-S\GS\S\GRS\:4GSjj55Gr{\R"\LGR5\C"5SSSSS\GRGS.S0\:SH\GRS-S\GRS-S\'S-S\GS\S\GRS\:4GSjj55Gr|\R"\LGRGR5\C"5SSSSSGS.SH\GRS-S\'S-S\GRS-GS\S\S\:4 GSjj55Gr}S\GRSSGS.S0\4SH\GRS-S\GRS\'S-S\S\:4 GSjjGr~GSGSL\7GS\\-\-GS\\-\-GS\7SH\GRS\'S\:4GSjjGr\R"\LGR5\C"5S0\:GS\:S1\;4GSj55Gr\R"\LGR5S0\:GS\:S1\;S\:4GSj5GrGSS0\:Sj\:S\S\S\S\4 GSjjGrS0\:Sj\:S\4GSjGr\R"\LGR5\C"SS9GSSGS.S"\:GS4\\-S-SG\*S-SL\SH\GRS-S\:4 GSjjj55Gr\R"\LGR 5\C"5SK\:S\:4GSj55GrGSGrG\"\5GrG\"\5GrG\"\5Gr\R"\LGR5\C"5GSS0\:GS\S\:4GSjj55Gr\R"\LGR5\C"5GSS0\:GS\S\:4GSjj55GrGS\GS\GSn\S\\\\44GSjGrS\GS\GS\SH\GRS\GRS\4 GSjGr\R"\LGR5\C"5GS\GR\GRGSSGS.GS\GS\GSn\SH\GRS\GRS\'S\S\:4GSjjj55GrGS\GS\GSn\S\\\\44GSjGr\R"\LGR"5\C"5GS\GR\GRGSSGS.GS\GS\GSn\SH\GRS\GRS\'S\S\:4GSjjj55Gr\R"\LGR$5\C"SS9SSGS.S0\;GS\:GS\GS\4GSjj55Gr\R"\LGR&5\C"5\A"SJ\,RS/9GSGSj555Gr\R"\LGR(5\C"5\A"SJ\,RS/9GSGSj555Gr\R"\LGR*5\C"5\A"SJ\,RS/9GSGSj555Gr\R"\LGR,5\C"5\A"SJ\,RS/9GS GSj555Gr\R"\LGR.5\C"5\A"GS\,RS/9GSSSSSSGS.GSjj555Gr\R"\LGR05GSQSGS.GSjj5Gr\U"\,R5SK\:4GSj5Gr\U"\,R5SK\:4GSj5Gr\R"\LGR65\C"5GSSG\*S-4GSjj55GrGSGrGSGr\R"\LGR<5\C"SS9G\\A"GS\,RS/9GS5555Gr\R"\LGR>5\C"SS9G\\A"GS\,RS/9GS5555Gr\R"\LGR@5\C"5S6\:GS\:SG\GS]\4GSj55Gr\W"\Y5Gr\W"\[5Gr\W"\\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\]5Gr\W"\^5Gr\W"\_5Gr\W"\`5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\b5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\c5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\e5Gr\W"\5Gr\W"\f5Gr\W"\g5Gr\W"G\[5Gr\W"G\\5Gr\W"G\5Gr\W"\h5Gr\W"\5Gr\W"\5Gr\W"\i5Gr\W"\k5Gr\W"\j5Gr\W"\l5Gr\W"\n5Gr\W"\m5Gr\W"\5Gr\W"\s5Gr\W"\5Gr\W"\5Gr\W"\t5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"G\q5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\~5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"G\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"G\5Gr\W"G\5Gr\W"\5Gr\W"\5Gr\W"\5Gr\W"G\5Gr\W"G\5Gr\W"G\5Gr\W"G\5Gr\W"\r5Gr\"\LGR5Gr\"\LGR5Gr\"\LGR5Gr\"\LGR45Gr\"\LGRB5Gr\"\LGR5Gr\"\LGRT5Gr\"\LGR5Gr\"\LGR5Gr\"\LGRzSGS9Gr \"\LGR5Gr \"\LGR5Gr GSGr GSGr GSGrGS\GRGS\;\<-4GSjGrGSGSjGrSSSSGS.GSjGrSSGKrSSGKrSSGKrSSGKrSSGKrg( N)CallableIterableSequence)Enum)partialreducesingledispatchwraps)AnycastOptionaloverloadUnion) sym_floatsym_intsym_maxsym_min)BoolLikeDeviceLikeTypeDimDimsSequenceTypeDimsType dtype_to_typeELEMENTWISE_TYPE_PROMOTION_KIND FloatLikeFloatWithoutSymFloatIntLike(is_contiguous_for_memory_format_or_falseis_contiguous_or_falseis_weakly_lesser_typeNumber NumberTypeRealNumberTypeREDUCTION_OUTPUT_TYPE_KIND ShapeType StrideType TensorLikeTensorLikeTypeTensorOrNumberLikeTypeTensorSequenceType)_maybe_convert_to_dtype_maybe_resize_out_safe_copy_out"elementwise_type_promotion_wrapper elementwise_unary_scalar_wrapper out_wrapper)highest_precision_float(absacosacoshasinhasinatanatanh bitwise_notceil conj_physicalcoscosh count_nonzerodeg2raddigammaerferfinverfcexpexpm1 exponentialexp2fillfill_floorfrac geometric index_add index_copy index_copy_ index_select index_fill index_fill_isfiniteisinfisposinfisneginfisnanisreali0lerplgammaloglog1plog2log10 log_normal log_softmaxmvlgammanormnormal nan_to_numnegpositiverad2deg reciprocalroundsigmoidsgnsignsignbitsinsincsinhsoftmaxsqrtsquaretantanhtracetruncaddatan2 bitwise_andbitwise_left_shift bitwise_orbitwise_right_shift bitwise_xor clamp_min clamp_maxcopysigndiveq float_power floor_dividefmaxfminfmodgcdgegt heavisidehypotigammaigammacimagiscloselcmle logaddexp logaddexp2 logical_and logical_not logical_or logical_xor logsumexpltmaximumminimummulne nextafterpowrealrpow remainderrsubrtruediv rfloordivsub true_divide trunc_dividexlogyaddcdivaddcmulclamp masked_fill masked_fill_whereclonecopy_toitemtoallamaxaminanycumsumcumprodmeandotvdotstdstd_meansum sum_to_sizeprodvarvar_meanaddralias alias_copy atleast_1d atleast_2d atleast_3d as_stridedas_strided_copyas_strided_scatter block_diagbroadcast_shapesbroadcast_tensors broadcast_tocatchunk column_stackconjconstant_pad_nd contiguous diag_embeddiagdiagonal diagonal_copydiagonal_scatterdsplitdstackexpand expand_as expand_copyflattenflipfliplrflipudhsplithstackmeshgridmovedimnarrow narrow_copynative_group_normnative_layer_normpermute permute_copyravelrepeatreshape reshape_asrollrot90rsqrtsplit_with_sizesstack swap_axessqueeze squeeze_copytt_copyTtake_along_dim tensor_split transposetranspose_copy unbind_copyunfold unfold_copy unsqueezeunsqueeze_copyviewview_as view_copyvsplitvstackview_as_complex unflattenunbindtriutril triu_indices tril_indicesarangecauchyempty empty_likeempty_permuted empty_stridedeyefull full_likelinspacelogspace new_emptynew_empty_stridednew_fullnew_ones new_zerosones ones_likerandn scalar_tensorzerozeros zeros_likeallcloseequal bucketize is_complexrenormstftistftc6USL=(d URS:g$)Nhputypedevices N/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_refs/__init__.pyis_noncontiguous_supportedr;qs T> 1V[[E11cSnSSKJn UH!n[XC5(dMURn O [ U5(dUR 5nU$)Nr) FakeTensor)torch._subclasses.fake_tensorr> isinstance fake_devicer;r) input_tlistoutputr9r>rs r:handle_noncontiguous_outputsrDusK F8  a $ $]]F  &f - -""$ Mr<c ^ ^ ^ ^ SSKJnJnJnJn [ R RRRn[S[SU555n[U5S:XagUH&m [T [5(aM[ST 35e S/[!["SU55-m [%U5GHum m ['SS[T 5- S5GH^m U"T T 5(a)U"T T 5(aU"T T T T :H5(aM<OU(aU"T T 5(avU"T T 5(afU"T T 5nU"T T 5nUS:Xa"US:wa[ R("T T S:H5 US:Xa"US:wa[ R("T T S:H5 U"T T T T :H5(aMU"T T S:H5(aT T S:a [+S 5eT T T T 'U"T T 5(dU"T T S:H5(aGM7[ R("T T T T :HU U U U 4S j5 GMa GM T $) Nr)guard_or_falseguarding_hint_or_throwhas_guarding_hint is_nested_intc3V# UHn[U[5(aU4OUv M! g7fNr@r.0xs r: $_broadcast_shapes..s*9A1g&&A-9s')c USL$rKrOs r:#_broadcast_shapes..s!4-r<zMInput shapes should be of type ints, a tuple of ints, or a list of ints, got c38# UHn[U5v M g7frKlen)rNshapes r:rPrQs4VESZZVz9Attempting to broadcast a dimension with negative length!c ,>STTSTSTSTST3 $)Nz.Attempting to broadcast a dimension of length z at z ! Mismatching argument at index z had z0; but expected shape should be broadcastable to rS)arg_idx common_shapeidxr[sr:rUrVs:LUSVZLX\]`\ab55Ar<)%torch.fx.experimental.symbolic_shapesrFrGrHrItorchfx experimental_configbacked_size_oblivioustuplefilterrZr@r RuntimeErrorrmax enumeraterange_check ValueError) _shapesrFrGrHrI backed_soshapesabr_r`rar[s @@@@r:_broadcast_shapesrusS%%--CCI /9F  6{a%**_`e_fg  .s4V45.6L$F+R#e*_b1CU3Z((!c!233#J,s"3399)%*55),s*;<<.uSz:A.|C/@AAAv!q& U3Z1_5Av!q& \#%6!%;<!%* S0A"ABBl3/1455:>$S%*#J S! s,,c a1P1P  %s3A[2,j r<T)preserve_cpu_scalar_tensorsc^^^^[SU56mS[S[S[4SjmUUU4Sjm[UU4SjU55$)Nc3h# UH(n[U[5(a UROSv M* g7frK)r@r'r[rNrs r:rP#_maybe_broadcast..s% G$QZ:..!''D 8$s02rsrtreturnc6SSKJnJnJn [ U5[ U5:wag[ X5HgupVU"XV:g5(a gU"U"US:HUS:H55(aOU"U"US:HUS:H55(a g[ R"XV:HS5 Mi g)Nr)rFsym_andsym_orTrWcg)NzCsizes assumed to be the same due to unbacked broadcasting semanticsrSrSr<r:rU9_maybe_broadcast..should_expand..s]r<F)rbrFr}r~rZziprcrn)rsrtrFr}r~rOys r: should_expand'_maybe_broadcast..should_expands  q6SV IDAaf%% ga1fa1f566qAvqAv 677 LL] (r<cH>Ucg[U[5(aU$[U[5(aOT(a[R"U5(aU$T"UR T5(aUR T5$U$[S[[U55-S-5e)Nz#Unexpected type when broadcasting: !) r@r!r'utilsis_cpu_scalar_tensorr[rrjstrr7)rOr[r`rvrs r:__maybe_broadcast+_maybe_broadcast..__maybe_broadcasts 9 6 " "H : & &*u/I/I!/L/LQWWl33xx --H5DG DsJ r<c36># UHnT"UT5v M g7frKrS)rNrOrr`s r:rPrzsBT"1l33Ts)rur%boolrh)rvargsrr`rs` @@@r:_maybe_broadcastrsH$ G$ GLyT@$ BTB BBr<)register_decompositionF)aten_op extra_meta exact_dtyper{c0^^^^S[4UUUU4SjjnU$)Nprimc>^[T5[TS9[[STS9S[S[4UU4Sjj5555nT[ La![ R"TTR5mTb[T5"U5 U$)Nrrstype_promoting_argstype_promotion_kindrsr{cD>TbT"U5 T"U5n[U/U5$rK)rD)rsrCrrs r:_ref>_make_elementwise_unary_reference..inner.._ref*s*%1 !WF/V< .inner's t  - ) + & 3  =N =~ =   * .  = m #''dmmT"U0UD6$rKrS)rkwargsfns r:_fn_make_alias.._fnJs4"6""r<r)rinspect currentframef_back f_globals __module__)rnamers` r: _make_aliasrBs5#L))+22<T"U/UQ7SU0UD6$)NoutrS)rsrrrs r:r_make_inplace.._fnYs!,d,,V,,r<_r) getmodule) r rrgetattratenrr__all__append)rr inplace_namer_alls` r: _make_inplacerRss 2Y--kk]!$LL |!< =c BC" R= D L! Jr<rc.[R"U5$rK)primsr2rs r:r2r2ks 99Q<r<c.[R"U5$rK)rr3rs r:r3r3s ::a=r<c.[R"U5$rK)rr4rs r:r4r4x ;;q>r<c.[R"U5$rK)rr6rs r:r6r6}rr<c.[R"U5$rK)rr5rs r:r5r5rr<c.[R"U5$rK)rr7rs r:r7r7rr<c.[R"U5$rK)rr8rs r:r8r8rr<c.[R"U5$rK)rr9rs r:r9r9s   Q r<c.[R"U5$rK)rr:rs r:r:r: ::a=r<inputcB[R"UR5$rK)ris_complex_dtypedtypers r:r0r0s  ! !%++ ..r<c|[R"UR5(dU$[R"U5$rK)rrrrr;rs r:r;r;s/  ! !%++ . .   u %%r<c.[R"U5$rK)rr<rs r:r<r< 99Q<r<c.[R"U5$rK)rr=rs r:r=r=rr<c.[R"U5$rK)rr@rs r:r@r@ == r<c.[R"U5$rK)rrArs r:rArArr<c.[R"U5$rK)rerf_invrs r:rBrBrr<c.[R"U5$rK)rrCrs r:rCrCrr<c.[R"U5$rK)rrDrs r:rDrDrr<c.[R"U5$rK)rrErs r:rErErr<c.[R"U5$rK)rrGrs r:rGrGrr<rrrsvaluec[U[5(d[S[U535e[U[5(d[S[U535e[ R "UR5n[ R"[U5U5(dS[U5SUS3n[U5e[R"X5$)Na must be TensorLike, got zvalue must be Number, got value argument of type  cannot be safely cast to type r) r@r'AssertionErrorr7r!rrrr rorrH)rsr python_typemsgs r:rHrHs a $ $9$q'CDD eV $ $9$u+GHH%%agg.K  & &tE{K @ @'U }4ST_S``abo ::a r<c^[R"X5n[R"X5 U$rK)rrHr)rsrrs r:rIrIs! 1A MM! Hr<c.[R"U5$rK)rcr,rs r:r*r*s   E ""r<c.[R"U5$rK)rrJrs r:rJrJ ;;q>r<rOc[R"[R"[R"U55[R"U55n[R "X5$rK)rcrrJr2rmr)rOtrunc_xs r:rKrKs< ii EIIaL15::a=AG 99Q  r<c[U[5(d[S[U535e[R "[ R"UR5S5 [R"U5$)Nrcg)Nz#imag only supports complex tensors.rSrSr<r:rUimag..s1Vr<) r@r'rr7rcrnrrrrrrs r:rr sW a $ $9$q'CDD LL qww')V ::a=r<)rc[R"UR5(d%[R"UR5(a[R "U5$[ U[RS9$)Nr) ris_float_dtyperrrrSr'rcrrs r:rSrSsM  AGG$$(>(>qww(G(G~~a  Qejj ))r<c[R"UR5(aQ[R"[ [R "U55[ [R"U555$[R"UR5(a"[R"U5[S5:H$[R"U[RS9$)Ninfr) rrrrcrrTrrrr2floatr,rrs r:rTrTs agg&&ejjm 4eEJJqM6JKK AGG$$yy|uU|++   AUZZ 00r<c,^[R"[R"TR5(+U4Sj5 [R "TR5(aT[ S5:H$[R"T[RS9$)Nc">STR3$)Nz7Complex dtype is not supported for isposinf, got dtype rrsr:rUisposinf../I!''Sr<rr rcrnrrrrrr,rrs`r:rUrU(sf  LL  " "177 ++S AGG$$E%L    AUZZ 00r<c,^[R"[R"TR5(+U4Sj5 [R "TR5(aT[ S5:H$[R"T[RS9$)Nc">STR3$)Nz7Complex dtype is not supported for isneginf, got dtype rrsr:rUisneginf..=rr<z-infrrrs`r:rVrV6sf  LL  " "177 ++S AGG$$E&M!!   AUZZ 00r<c.[R"X5$rKrrrs r:rWrWDs 88A>r<rbc[R"UR5(a[R"U5S:H$[R "U[R S9$)Nrr)rrrrcrr'rrs r:rXrXMsA  agg&&zz!}!! ??1EJJ //r<c.[R"U5$rK)r bessel_i0rs r:rYrYXs ??1 r<c.[R"U5$rK)rr[rs r:r[r[_s <<?r<c.[R"U5$rK)rr\rs r:r\r\drr<c.[R"U5$rK)rr]rs r:r]r]irr<c.[R"U5$rK)rr^rs r:r^r^nrr<c.[R"U5$rK)rr_rs r:r_r_srr<dimrc U=(d URn[R"U5n[X5n[U[ XQSS9- U5$)NTkeepdim)rrget_computation_dtyper+r)rsr r result_dtypecomputation_dtypea_s r:raraysF #AGGL33LA  6B "2 "4(H#H, WWr<selfrr cX[U[5(dU4nUR5S:Xa9[R"[R "U5X5R 5$[R"[R"U5USS9n[R"X3R5[S5:HS5nU(aUO[R"X15n[R"[R "X- 5X5nUR 5RU5$)NrTr r)r@rnumelrcrrDr\rrrr2rrry)rr r maxesmaxes_squeezedresults r:rrs c8 $ $f zz|qyy4#7;;== JJuzz$'d ;E   eYY[E%L%@! DE%U5==+DN YYuyy. =F ::<  N ++r<nanposinfneginfc[U[5(d[S[U535e[R "UR 5(d%[R"UR 5(aUR5$UcSnUc*[R"UR 5RnUc*[R"UR 5Rn[R"[R"U5X5n[R"[R"U5X45n[R"[R "U5X$5nU$)Nr)r@r'rr7ris_boolean_dtyperis_integer_dtyperrcfinforkminrrWrVrU)rsrrrrs r:reres a $ $9$q'CDD agg&&%*@*@*I*Iwwy { ~QWW%)) ~QWW%)) [[Q 0F [[*F ;F [[*F ;F Mr<ch[R"UR[RLS5 g)Ncg)NzNegation, the `-` operator, on a bool tensor is not supported. If you are trying to invert a mask, use the `~` or `logical_not()` operator instead.rSrSr<r:rU_neg_meta..s r<)rcrnrrrs r: _neg_metar$s# LL uzz! r<)rc.[R"U5$rK)rrfrs r:rfrfs 99Q<r<c[U[5(d[S[U535eUR[ R La Sn[U5eU$)Nrz'positive does not support bool tensors.)r@r'rr7rrcrrj)rsrs r:rgrgsK a $ $9$q'CDDww%**73 Hr<c[U[5(d[S[U535e[R "UR 5(a[R"U5$U$)Nr) r@r'rr7rrrrrrs r:rrsN a $ $9$q'CDD agg&&zz!} Hr<c.[R"U5$rK)rrirs r:riris   A r<)decimalsr)cUS:Xa[R"U5$SU-nSU*-n[R"[R"[R"X55U5$)Nr )rrjr)rsr)ten_pow ten_neg_pows r:rjrjsQ1}{{1~h,hY' yyUYYq%:;[IIr<c.[R"U5$rK)rrrs r:rrrr<c R[S[S[[U5555$NrW)rryrDrfrs r:rkrks q#aSV- ..r<c[R"UR5(a-UR5n[R "US:HSX- 5$UR 5$Nr)rrrr2rcrrm)rsa_abss r:rlrlsH  agg&&{{5A:q!)44vvxr<c.[R"U5$rK)rrmrs r:rmrm rr<c.[R"U5$rK)rrnrs r:rnrns == r<c.[R"U5$rK)rrors r:rororr<c[RU-n[R"US:HS[R"U5U- 5$NrrW)mathpircrrors r:rprp s3 ! A ;;qAvq%))A,"2 33r<c.[R"U5$rK)rrqrs r:rqrq&rr<c.[R"U5$rK)rrsrs r:rsrs+rr<c[X5$rK)rrs r:rtrt0s q9r<c.[R"U5$rK)rrurs r:ruru8rr<c.[R"U5$rK)rrvrs r:rvrv=rr<c.[R"U5$rK)rrxrs r:rxrxBrr<c^URm[R"[R"T5U4Sj5 UR 5n[R"[ U5S:gS5 [R"USS:HS5 UR5n[R"USS:HS5 USSn[R"[R"S U55S 5 [R"UR5S-S:HS 5 [R"U[R"T55RS5$) Nc>ST3$)Nz^view_as_complex is only supported for floating pointtensors, but got a tensor of scalar type: rS) input_dtypesr:rU!view_as_complex..Os55@MCr<rcg)Nz-Input tensor must have one or more dimensionsrSrSr<r:rUrDU?r<r]cg)Nz+Tensor must have a last dimension of size 2rSrSr<r:rUrDY=r<rWcg)Nz/Tensor must have a last dimension with stride 1rSrSr<r:rUrD_Ar<c30# UH oS-S:Hv M g7f)rGrNrS)rNstrides r:rP"view_as_complex..cs84aZ1_4scg)NzCTensor must have a stride divisible by 2 for all but last dimensionrSrSr<r:rUrDdsUr<cg)Nz0Tensor must have a storage_offset divisible by 2rSrSr<r:rUrDhsBr<)rrcrnrrsizerZrMbuiltinsrstorage_offsetrview_element_typecorresponding_complex_dtyper)rsizes old_stridesdimsrCs @r:rrKs**K LL [) C IIKE LL E a? LL b Q= ++-K LLB1A s D LL 8488U LL !Q&B  " " e// <  gbkr<c @^^^^^^^^S[4UUUUUUUU4SjjnU$)Nrch>^Tc TRm[T5[ST S9S[[-S[[-S[4UUUUU4Sjj55nT(a[ 5"U5nTUlT[ La[R"TT5mTbT(a[T5"U5 U$)Nrsrtrrsrtr{c>[R"T=(d [U[5(+U4Sj5 [R"T=(d [U[5(+U4Sj5 [R"T=(d, [U[5=(a [U[5(+U4Sj5 [ X5upT"X5n[ X/U5$)Nc>TS3$)Nzc: Received a lhs Python scalar to an elementwise binary operation that does not accept lhs scalars!rSrsr:rUQ_make_elementwise_binary_reference..inner.._ref..4&!>>r<c>TS3$)Nzc: Received a rhs Python scalar to an elementwise binary operation that does not accept rhs scalars!rSr^sr:rUr_r`r<c>TS3$)Nz?: Receive two Number inputs to an elementwise binary operation!rSr^sr:rUr_s 4& _`r<)rc _check_valuer@r!rrD)rsrtrCrrsupports_lhs_python_scalarsupports_rhs_python_scalarsupports_two_python_scalarss r:r?_make_elementwise_binary_reference..inner.._ref~s   *G*Q2G.G>    *G*Q2G.G>    +I"1f-G*Q2GH`  $A)DA!ZF/? ?r<) rr r.Tensorr"r0rrrr) rrrhas_outrshould_register_decompositionrdrerfrs ` r:r1_make_elementwise_binary_reference..innerys <==D t + * 3  @ " @ " @ @ @     @. =&D m #''d3G  #@ "7 +D 1 r<r) rrrrirdrerfrjrs ```````` r:"_make_elementwise_binary_referencerlos*H**X Lr<rtc[U[5(a UR$[U[5(a UR$[R"[ U55$rK)r@r'rr type_to_dtyper7r[s r:_binary_op_dtyperosE!Z  ww!Z  ww   tAw ''r<r[alpharqc[X5upUb[X5n[R"U5nU[LaB[R "[ U5U5(dS[ U5SUS3n[U5e[U[5(a[R"X5nOX-n[R"X5n[X/U5$)z' Reference implementation of torch.add alpha argument of type rr)rrorrrr r7ror@r'rrryrDrsrtrqrrrrCs r:ryrys A !DA  &))%0 d "5+F+F K, , ,DK=8WXcWddefCS/ ! a $ $ !#A A YYq_F ' 77r<)rrdrec.[R"X5$rK)rrzr[s r:rzrz ;;q r<)rc.[R"X5$rK)rr{r[s r:r{r{   Q ""r<c.[R"X5$rK)r shift_leftr[s r:r|r|   A !!r<c.[R"X5$rK)rr}r[s r:r}r}r{r<c.[R"X5$rK)rshift_right_arithmeticr[s r:r~r~s  ' ' --r<c.[R"X5$rK)rrr[s r:rrrxr<)rrdc[U[5(a4[U[5(a[XRUR S9nOl[U[5(aW[U[5(aBUR UR :wa(SUR SUR S3n[ U5e[[U5[[U55[U55$)Nrr9/Expected divisor (b) to be on the same device (&) as dividend (a), but it is found on r) r@r!rhr)rr9rjrrnrfr2)rsrtrs r:rrs !VAv!6!6 !77188 < Av  :a#8#8QXX=Q?zIopqpxpxoyyz{3 SQ[#a& 11r<) rounding_modercUc [X5$US:Xa [X5$US:Xa [X5$SUS3n[U5e)z' Reference implementation of torch.div rxrJzLdiv expected rounding_mode to be one of None, 'trunc', or 'floor' but found .)rrrro)rsrtrrs r:rrsV1  ' !A!! ' !A!!\]j\kklmor<c.[R"X5$rK)rrr[s r:rr* 88A>r<cl[U[5(d [U[5(d [S5e[U[5(a=US:XaUR 5$US:XaX-$US:Xa[ R "U5$O[U[5(aUS:Xa[ R"US5$US:Xa`[R"UR5(d%[R"UR5(a[ R"U5$[R"X5$)Nz-at least one of a or b must be TensorLikeType?g@?T)r@r(rr!rrcrsrHrrrrrGrrr[s r:rr2s q. ) )Z>-J-JLMM!V 8779  #X5L #X::a=  Av   8::a& & 8   ) )U-C-CAGG-L-L::a= 99Q?r<c[U[5(a [U[5(a [S5e[R"X5nUc [ S5e[R "U5(a[RnO[Rn[X5n[X5n[X5up[X5$)Nz=Receive two Number inputs to an elementwise binary operation!z/dtype should not be None after get_higher_dtype) r@r!rorget_higher_dtyperrrc complex128float64r+rrrsrtrs r:rrPs !VAv!6!6 K   " "1 (E }NOO e$$    )A)A A !DA q9r<)rrfrjcZ^[U[5(a-[U[5(a[U5n[U5nGO@[U[5(a4[U[5(a[XRUR S9nO[U[5(a4[U[5(a[XRUR S9nO[U[5(a[U[5(aUR UR :wajUR [ R "S5:Xa(SUR SUR S3n[U5e[R"XR S9n[U[5(a[U[5(d [S5eURm[R"T5(a [X5$[R"T5(a [X5$[ R "SU4S j5 g) Nrcpurrrr8z*a and b must both be Tensors at this pointFc>TS3$)Nz not supported for floor_dividerSrsr:rUfloor_divide..s ug-L$Mr<)r@r!r)rhrr9rcrjr device_putrrr_floor_divide_floatr_floor_divide_integerrn)rsrtrrs @r:rrs!VAv!6!6 !  !  Av  :a#8#8 !77188 < Av  :a#8#8 !77188 < Av  :a#8#8QXX=Q 88u||E* *CAHH:Mstut|t|s}}~Cs# #  884A q& ! !jF&;&;IJJ GGE E"""1((    & &$Q** UMNr<c|[X5upURR(d[R"X5$[ R "U5[ R "U5:gR[ R"X5S:g5n[R"X5[X R5- $r2) rr is_signedrrrcrnrrr+)rsrtoffsets r:rrs~ A !DA 77  yymmA%--"22 ? ? 1@PTU@U VF 99Q?4VWWE EEr<c [X5n[[X5U5n[[ US5[ US55n[ US5n[ XT5n[U[US5U5n[U5n[[X75S5n[U[US5U5n[X5n[SURURS9n [[ US5U[X55n[[ US5Xx5$)NrrWrr)rrrrrrr{rrJrryr)rr9r) rsrtmodrdifferent_signed_inputsnon_zero_remaindermask floor_div basic_div zero_tensors r:rrs q*C c!k1 %C*"Q(Bq!H=C ) CD c#qk3 'Cc I c#!3 'DdC 1-y9IA!IAQAQRKbaj)Xk-MNI Aq9 00r<c.[R"X5$rK)rrr[s r:rr ::a r<c.[R"X5$rK)rrr[s r:rrrr<c.[R"X5$rK)rrr[s r:rrrr<mantissaexponentch[RR[R"U55$rK)rc return_typesfrexprrs r:rrs$    # #EKK$5 66r<c.[R"X5$rK)rrr[s r:rrs 99Q?r<c.[R"X5$rK)rrr[s r:rrrr<c.[R"X5$rK)rrr[s r:rrrr<valuesc[R"US5n[R"[R"US5[R"U55n[R "USS5n[R "X!U5nU$r8)rcrrrrWr)rr input_eq_zero input_lt_zerozeros_and_onesrCs r:rr sa HHUA&M$$UXXeQ%7U9KLM[[15N [[ ?F Mr<c.[R"X5$rK)rrr[s r:rrrvr<c.[R"X5$rK)rrr[s r:rr#s << r<c.[R"X5$rK)rrr[s r:rr,s == r<rrtolatolc^^^^^[R"TRTR:HUUU4Sj5 [R"TS:UU4Sj5 [R"TS:UU4Sj5 g)NcB>TSTRSTRS3$)Nz4: Attempting to compare tensors of different dtypes  and rr)rsrtrsr:rU#_check_close_args..>s'4&LQWWIUZ[\[b[bZccder<rc>TSTS3$)Nz6: rtol must be greater than or equal to zero, but got rrS)rrsr:rUrB4&NtfTUVr<c>TSTS3$)Nz6: atol must be greater than or equal to zero, but got rrS)rrsr:rUrFrr<)rcrcrrnrrsrtrrs`````r:_check_close_argsr5sV  177e LL  V LL  Vr< equal_nanc 0[SXX#S9 [X5nU(ar[R"UR5(d%[R "UR5(a([ U[[U5[U555nUS:XaUS:XaU$[R"UR5(dy[R "UR5(dT[R"U[R"55n[R"U[R"55n[U[[X555n[[!X55n[ U[[#U5[%Xv555nU$)Nz torch.iscloserr)rrrrrrrrrWrconvert_element_typercget_default_dtyperyr2rrrSr) rsrtrrrclose allowed_error actual_errorrs r:rrKs?a4K qHEe**17733u7M7Magg7V7V5+eAha"AB qyTQY     ( (1G1G1P1P  & &q%*A*A*C D  & &q%*A*A*C Dc#a,/0Ms1y>L {8L12l3RSF Mr<cURnU[R[R4;nU(aJ[R "U[R 5n[R "U[R 5n[R"X5n[R"US:HSU5n[R"[R"X5U-5nU(dU$[R "XR5$r8) rrcint8int16rrint32rrr2r)rsrtrpromote_to_intgress r:rrvs GGEuzz5;;77N  & &q%++ 6  & &q%++ 6 !A AFAq!A ))EIIaOa' (C$3P%*D*DS*PPr<c.[R"X5$rK)rrr[s r:rrrr<c [R"U5[R"U5:n[R"X U5n[R"X!U5n[R"[R"[R "[R"U555[R"U5[R"U5:H5n[ R"UR5(d%[ R"UR5(a[R"U5S:n[R"Xd[R"[R"U5[R"U5-55n[R"XWU[R"[R"XC- 55-5n[R"U5n [R"U [[S5[S55U5$[R"XPU[R"[R"XC- 55-5$)Nrr)rcrrrrrSrrrr\rDr]rWcomplexr) rsrtrmax_min_inf_mask neg_min_maskinf_vals non_nan_valsnan_masks r:rrs~ ::a=EJJqM )D ;;t "D ;;t "D   %..A78%**Q-5::VW=:XH agg&&%*@*@*I*Izz$'!+ ;;  %))D/EIIdO*K L {{ u{{599T[3I'J J ;;t${{8WU5\5<%H,WW{{8u{{599T[;Q/R(RSSr<cH[R"[R"UR5=(d [R"UR5(+S5 X:n[R "X U5n[R "X!U5n[R "[R"U5X:H5nS[R"S5- nU[R"[R"XC- 55U--n[R "XPU5$)Ncg)Nz)logaddexp2 doesn't support complex dtypesrSrSr<r:rUlogaddexp2..;r<rrG) rcrnrrrrrrTr9r\r]rG)rsrtrrrr inv_log_2rs r:rrs  LL  # #AGG , O0F0Fqww0OP; 6D ;;t "D ;;t "D  Q8Hdhhqk!I EKK 4; 789D DF ;;xF ++r<c[R"UR5(dUS:gn[R"UR5(dUS:gnX-$r2rrrr[s r:rrG  ! !!'' * * F  ! !!'' * * F 5Lr<c\[R"UR5(dUS:H$U)$r2rrs r:rrs'  ! !!'' * *Av 2Ir<c[R"UR5(dUS:gn[R"UR5(dUS:gn[X5$r2)rrrr}r[s r:rrsK  ! !!'' * * F  ! !!'' * * F a r<c[R"UR5(dUS:gn[R"UR5(dUS:gnX- $r2rr[s r:rrrr<c.[R"X5$rK)rrr[s r:rrrr<c.[R"X5$rK)rrr[s r:rr == r<c.[R"X5$rK)rrr[s r:rrrr<)rrfc.[R"X5$rK)rrr[s r:rrs 99Q?r<c.[R"X5$rKrr[s r:rr rr<c.[R"X5$rK)rrr[s r:rrs ??1  r<c.[R"X5$rK)rrr[s r:rrs ??1  r<rWcp[U[5(a Sn[U5e[R"XUS9$)Nz?Received a Number for the first argument, but expected a Tensorrp)r@r!rorcr)rsrtrqrs r:rr$s1!VOo 99Q ''r<c[X5up[U[5(a|[U[5(ag[R"[ R "UR5(+=(a% [ R "UR5(+S5 US:wa[X5n[ R"U5n[ R"[U5U5(dS[U5SUS3n[U5e[U[R5(a[R"X5nOX-n[R "X5n[#X/U5$)z' Reference implementation of torch.sub cg)NzwSubtraction, the `-` operator, with two bool tensors is not supported. Use the `^` or `logical_xor()` operator instead.rSrSr<r:rUsub..IsCr<rWrsrr)rr@r'rcrnrrrrorr r7rorhrrrrDrts r:rr4s  A !DA!Z  Z:%>%> &&qww/ / W8N8Nqww8W4W   z &))%0 **4; DD+DK=8WXcWddefCS/ ! a & & !#A  A YYq_F ' 77r<r)rrrrfc.[R"X5$rK)rrr[s r:rras 99Q?r<c d[R"[U[5=(d [U[5S5 [U[5(a4[U[5(a[ XR URS9nOH[U[5(a3[U[5(a[ XR URS9n[U[5(d[S[U535e[U[5(d[S[U535e[R"[R"US5S[R"U[R"U555n[R"[R"U5[S5U5$)Ncg)Nz/Expected either argument a or b to be a Tensor"rSrSr<r:rUxlogy..trKr<rrzb must be TensorLike, got rr)rcrnr@r'r!r)rr9rr7rrrr\rWr)rsrtrhss r:rrks LL1j!>Z:%>A !Z  Z6%:%: !77188 < Az " "z!V'<'< !77188 < a $ $9$q'CDD a $ $9$q'CDD ++ehhq!na1eiil)C DC ;;u{{1~uU|S 99r<)rrrfc[R"U5n[R"U5(a[R"X5$[ [R"X55$rK)r get_dtyperrrrxrs r:rrsD OOA E e$$yy 1 !!r<)rtensor1tensor2rrrc^^Tb\URn[R"U5m[R"[R "[ T5T5UU4Sj5 UTU-U- -$)z+ Reference implementation of torch.addcdiv c(>S[T5STS3$Nrrrr6rrsr:rUaddcdiv..-d5k]:YZeYffghr<rrrrcrcr r7rrrrrrs ` @r:rr_   ))%0    ' 'U [ A h %'/G+ ++r<c^^Tb\URn[R"U5m[R"[R "[ T5T5UU4Sj5 UTU-U--$)z+ Reference implementation of torch.addcmul c(>S[T5STS3$rr6rsr:rUaddcmul..rr<rrs ` @r:rrrr<)rsr rkr rkcUcUc Sn[U5eUbX[R"U5n[R"[R"X5U5n[R "XPU5nUbX[R"U5n[R"[R "X5U5n[R "XPU5nU$)Nz+clamp called but both min and max are none!)rorcrWr}rrr)rsr rkra_isnan conditions r:rrs {s{;o ++a.$$UXXa%5w? KK c * ++a.$$UXXa%5w? KK c * Hr<c*[R"XS9$)N)r rcr)rr s r:rr ;;t %%r<c*[R"XS9$)N)rkr )rrks r:rrr r<pred.c,[R"USS9$)NT)as_tuple)rcnonzerors r:_where_defaultr s == --r<c^UbUc[e[R"TXSS9 [R"TR [R LU4Sj5 [TX5ump[R"TX5$)Tallow_cpu_scalar_tensorsc">STR3$)Nz#expected predicate to be bool, got rrsr:rUwhere..%s5djj\Br<) NotImplementedErrorrcheck_same_devicercrnrrrrr)rrsrts` r:rrsh" yAI!! D!F LL ejj B "$-JD! ;;tQ ""r< memory_formatrc.[R"XS9nU$Nr)rr)rsrrs r:rr/s [[ 8F Mr<)allow_cross_devicecU(dBURUR:wa(SURSURS3n[U5e[R"X5$)NzAttempting to copy from device z to device z*, but cross-device copies are not allowed!)r9rjrr)rsrtr rs r:rr8sK !((ahh"6/zQXXJNxy3 == r<cUR5S:waSUR5S3n[U5e[R"UR5nU"[ R "U55$)NrWzCan't convert a tensor with z elements to a number!)rrorrrrr)rsr number_types r:rr@sWwwyA~,QWWYK7MNo%%agg.K uzz!} %%r<r9copylayout pin_memory non_blockingcVU(+=(a USL=(d URU:H=(a{ USL=(d URU:H=(aZ USL=(d URU:H=(a9 USL=(d. U[R:H=(d [ R "XS9$r)r9rr%rcpreserve_formatris_contiguous_for_memory_format)rsr9rr$r%rr&r's r:_to_will_aliasr+Ms  t^ 1qxx61 d] .agg. t^ 1qxx61  T ! U 5 55 U44QT r<c[erK)r)rrs r: _to_dispatchr-fs r<cUUUUUS.nU$N)r9rr'r$rrSr9rr'r$rrs r: _to_devicer1ks!$& F Mr<c>[R"U5UUUUS.nU$r/)rcr9r0s r:_to_device_strr3}s*,,v&$& F Mr<cUUUUS.nU$)N)rr'r$rrS)rr'r$rrs r: _to_dtyper5s$& F Mr<otherc`URnURnURnUUUUUUS.nU$)N)r9rr%r'r$r)r9rr%)r6r'r$rr9rr%rs r: _to_otherr8sA\\F KKE \\F$& F Mr< to_kwargsc/SQnSU;a4[US[5(a[R"US5US'UHnX1;dM US:XaX[RLdyUS:Xa_XR URR :Xa9XR (a:XR URR :Xd[XS5X:XdMURU5 M g)N)rr9r%rr9r) r@rrcr9r)r7indexrpop)rsr9options_to_checkkws r:_canonicalize_to_argumentsr?sE9Ih,?!E!E#ll9X+>? ( ?&9=E# UHnTRUS:v M g7frNr[rNirss r:rP_reduction.. s.Lt!qwwqzA~ts"cg)NzJreducing over zero-size dimension for reduction operation without identityrSrSr<r:rU_reduction.. s`r<rWz4result_dtype should not be None when out is providedz7Expected the dtype of reduction result and out to match copy_fromr)r@r'rr7ndimrjrrrreduction_dimsr[rbr}rcrnreduction_dtypesr+rmrbroadcast_in_dimr,r-r)rsrrBrCrXrDrrrEr} valid_shaperrrrM output_shapebroadcast_dimss` r: _reductionrZsA a $ $9$q'CDDvv{%affX-l m   #z** #?S {!KL L   !"J  4 5 5 #A$t*!NO O$w    .D AffkMW.Lt.L%M   ` ',&<&< '# #45A !T]FDI!&&MRMqtm :M R%*166]D]tm!]D''nM    !WX X  !:I  \\2<< |||# (@++FA M#SDs( J! J .J cF^^[[TR5n[TS05n[SS9"U5m[ T5SS.UU4Sjj5nTRS3nXTlUR R U5 [[[U55"U5 U$)zc Given a view function (e.g. torch.diagonal) generates its copy variant (e.g. torch.diagonal_copy) __annotations__TrNrch>T"USU0UD6nT(aUbgUbU$[R"SU5$)Nrc<UR[RS9$r)rrccontiguous_formatrTs r:rU3_make_copy_from_view.._fn..P saggE,C,CgDr<)pytreetree_map)rrrrrreturn_none_on_out_variants r:r!_make_copy_from_view.._fnG sBT-s-f- %#/ ?M D   r<_copy)rrrr0r r\updater)rrdaten_fn annotationsr copy_names`` r:_make_copy_from_viewrk= sdBKK(G"/4K  &w /B 2Y     ;;-u%IL{+7434S9 Jr<c[R"[R"[R"U5XS95nUR[R:XaUR [RS9nU$)Nr r)rcrrruint8r)rsr r rs r:rr[ sS  uyy):):1)=sT UFww%++- Mr<cp[U[R5n[U[[ 45(a [ U5S:XaUR5nOURXS9RS5nUR[RLa%[R"U[R5$U$)Nr)r r F)r+rcrr@listrhrZrrrrrmrr)rsr r rrs r:rrj s !EJJ /B#e}%%#c(a-C144U; ww%++))&%++>> Mr<)rrc hUcwUb URnOg[R"UR5(d%[R"UR5(a[R nO URnUS:XdU/:XaSn[ U[RUUUU[RS9$NrSrXrDrrrE) rrrrrcint64rZrrr$SAMErsr r rrs r:rr~ s } ?IIE  # #AGG , ,0F0Fqww0O0OKKEGGE byC2I    499 r<c ^^^[R"TSS9m[R"[R"TTR 5UU4Sj5 [R "TTR 5(a%[T5S:a[R"T5$TR[T5- m[[T55[UUU4Sj[T[T5555-n[R"TUSSS9$)NFvalidatec*>STSTRS3$)Nzsum_to_size: size "z" is not expandable to size ""rKrsr[sr:rUsum_to_size.. s%eW,I!''RSTr<rc3n># UH*nTUT- S:XdMTRUS:wdM&Uv M, g7frWNrK)rNrMrs leading_dimsr[s r:rPsum_to_size.. s@50A \! "a ' ,-GGAJ!O 0s 55 5T)r r r)rextract_shape_from_varargsrcrnis_expandable_tor[ is_same_shaperZrview_ofrSrhrmr)rsr[ reduce_dimsrs`` @r:rr s  , ,UU CE LL uagg.T  5!''**s5zA~}}Q66CJ&Ll+,u5|SZ050K 99QKT BBr<c hUcwUb URnOg[R"UR5(d%[R"UR5(a[R nO URnUS:XdU/:XaSn[ U[RUUUU[RS9$rq) rrrrrcrsrZrrr$rtrus r:rr s } ?IIE  # #AGG , ,0F0Fqww0O0OKKEGGE byC2I    499 r<r]c vUS:XdU/:XaSn[U[RUUSUS[RS9$NrSFrXrDrrrBrE)rZrrr$rtrsr r rs r:rr E byC2I    499  r<c vUS:XdU/:XaSn[U[RUUSUS[RS9$r)rZrrr$rtrs r:rr rr<c@Uc[U[5(aUnSnX4$rK)r@r)r unbiaseds r:_dim_var_dispatchr s(JsD11 =r< correctionrrc [X5up[R"X$5nUS:XdU/:XaSn[U[ [ R US9UUSSS[RS9nU$)NrSrTr) rrset_correctionrZrrrr$COMPLEX_TO_FLOAT)rsr rr rrs r:rr sl&c4MC%%h;J byC2I   j1  4EE F Mr<c4[X5up[R"X$5n[R"U[R 5upV[ X5n[R"XXCS9n[R"U5nUc [S5e[ X5$N)rr z/dtype should not be None after reduction_dtypes) rrrrUr$rr+rcrrsr) rsr rr r opmath_dtypera_vara_stds r:rr' s&c4MC%%h;J00 % 6 6L 0A IIa EE JJu E }NOO "5 00r<c ^^^ US:XdU/:XaSnTm Tc TRm[T[RUUTS[R S9n[ R"[R"T5=(d [R"T5UU 4Sj5 [U[5(aU4n[R"TRU5nTRS:XaSO$[!["R$U4SjU5S5n['XW5nTc TROTn[)XX5nUbJ[U[*5(d[-S[/U535e[1XER5n[3XTS9$U$) NrSrrc>STcSOSST3$)Nz&mean(): could not infer output dtype. Inputr z> dtype must be either a floating point or complex dtype. Got: rS)r orig_dtypesr:rUmean..Y s"4$,w*=>77# UHnTRUv M g7frKrKrLs r:rPmean..b s7QDq DsrHrQ)rrZrrr$KEEP_PROMOTED_TYPErcrnrrrr@rrTr[rSroperatorrrr+r'rr7r,r-) rsr r rrrrXnelemrrs ` ` @r:rr? s? byC2IJ }    4GGF LL U#Du'='=e'D #sf    -D1A&7QD7QST"UE  'F#m177L $V :F #z** #?S {!KL L\\2<< Mr<out0out1)rr rcf[X5up[R"X$5n[R"U[R 5upVUR n[X5n[R"XXCS9up[R"U5n Uc [S5e[X5[X54$r) rrrrUr$rrr+rcrrsr) rsr rr rrroriginal_dtypera_meanrs r:rrn s&c4MC%%h;J00 % 6 6LWWN0ANN1jRME JJu E }NOO-7 r<cP[X5up[XX#US9n[XU5nXV4$)Nr)rrr)rsr rr rvms r:rr s1&c4MC AH*=A QWA 4Kr<)rvec1vec2)betarqrrrc^^^^^^[R"TRS:HU4Sj5 [R"TRS:HU4Sj5 TS4TS44Hunm[U[5(dM[R"[ R "TR5=(aG [ R "TR5=(a [ R "TR5U4Sj5 M TRTRSTRS5m[ R "TR5(a[R"[[T5[5U4Sj5 [R"[[T5[5U4S j5 T(d5T(a[R"TT5$[R"TS 5$[R"TT(a"TT55$[R"TS 55$[R"[[T5[!TR55UU4S j5 [R"[[T5[!TR55UU4S j5 TS:XaT[R"TT5-$TT-T[R"TT5--$) NrWc$>STRS3$)Nz*addr: Expected 1-D argument vec1, but got -DrS)rsr:rUaddr.. STRS3$)Nz*addr: Expected 1-D argument vec2, but got rr)rsr:rUr rr<rqrc>STS3$)NzBoolean z$ only supported for Boolean results.rS)arg_namesr:rUr s(8*,PQr<rc >S[T53$)Nzexpected bool/int beta but got r6)rsr:rUr s5d4j\Br<c >S[T53$)Nz expected bool/int alpha but got r6rpsr:rUr s6tE{mDr<Fc:>S[T5STR3$Nzcannot safely convert z to r7r)rrsr:rUr s,T$ZLTZZLIr<c:>S[T5STR3$rr)rqrsr:rUr s,T%[Mdjj\Jr<)rcrnrSr@rrrrrr[r r7intouterrrr)rrrrrqargrs````` @r:rr s LL QJ LL QJ!'*T6N; X c4 LL&&tzz27**4::67**4::6Q  < ;;tzz!}djjm 4D djj)) !$t*c 2 B   !$u+s 3 D .35;;tT* Uu9U U##+0 D$' 6;oodE6R   !$t*mDJJ.G H I   !$u+}TZZ/H I J  195;;tT22 2$;T4)@!@@ @r<rrc,U(d,[U[RR5(aUnO:[U[RR5(a [ S5eU4U-n[ SU55n[ U5S:aU$US$)z5Reference implementation of :func:`torch.atleast_1d`.2arg should not be a Sequence when args is providedc3\# UH"oRS:aUO [US5v M$ g7f)rWrN)rSrrNrss r:rPatleast_1d.. s#Eu!VVq[i1o5us*,rWr)r@ collectionsabcrrrhrZ)rrargs_rs r:rr sy JsKOO$<$<== c;??33 4 4 !UV V  EuE ECc(Q,3*CF*r< at_least_fncU"U5n[U[5(d[S[U535e[ X15$)Nz(at_least_fn must return TensorLike, got )r@r'rr7r)rr rarg_s r:_unsqueeze_atleastr s= s D dJ ' 'GT |TUU T r<c^^U(d,[U[RR5(aUnO:[U[RR5(a [ S5eU4U-n[ [ [S5m[U4SjU55n[U5S:aU$US$)z5Reference implementation of :func:`torch.atleast_2d`.rrc3V># UHoRS:aUOT"U5v M g7f)rGNr)rNrsunsqueeze_atleast_1ds r:rPatleast_2d.. $Mu!VVq[&:1&==u&)rW) r@rrrrrrrrhrZ)rrrrrs @r:rr s JsKOO$<$<== c;??33 4 4 !UV V "#5z1E MuM MCc(Q,3*CF*r<c^^U(d,[U[RR5(aUnO:[U[RR5(a [ S5eU4U-n[ [ [S5m[U4SjU55n[U5S:aU$US$)z5Reference implementation of :func:`torch.atleast_3d`.rr]c3V># UHoRS:aUOT"U5v M g7f)Nr)rNrsunsqueeze_atleast_2ds r:rPatleast_3d.. rrrWr) r@rrrrrrrrhrZ)rrrrrs @r:rr s JsKOO$<$<== c;??33 4 4 !UV V "#5z2F MuM MCc(Q,3*CF*r<rQrMrScZUbUOUR5n[R"XX$5$rK)rSrr)rsrQrMrSstorage_offset_ints r:rr s1)4!:J:J:L   AV @@r<srcc@UcSOUn[R"XX#U5$r2)rr)rrrQrMrSrs r:rr s'-4.  # #E>P QQr<c:[R"[U65$rK)rcSizeru)rrs r:rr, s ::'0 11r<c[U5S:Xa[US[5(dUSn[[ USS065$)NrWrrvF)rZr@rhrortensorss r:rr0 s? 7|qGAJ!?!?!*  'MuM NNr<c[U5[UR5- n[[U[UR5U-55n[R "XU5$rK)rZr[rhrmrrV)rsrQstartrXs r:rr9 sI IAGG $E uc!''lU23 4D  ! !!4 00r<rrcT^ ^ ^ ^ Sn[U5S:Xa Sn[U5eUH/n[U[5(aM[ S[ U535e [ R"USS06 SSKJ n Sm [U5Heum m T cT RS:waT m MMT RS:wdM1[R"T RT R:HU U U 4S j5 Mg T cUSm T Rn/n[U5Hum n[U5[UR5:wa\URS:wa[ S UR35e[R"U"URSS:H5U U 4S j5 MURS:XaU"URSS:H5(aMURU5 M U"U5n[U5S:XaBUSm [![#S U555n ['S T R(T R*U US9$[ R,"USRU5n[ R."USRU5 [0R2"Xq5R5US9$![$a Sn Nf=f)NcSnUHMn[R"U5nU[R:XaUs $UbX:wa[Rs $UnMO Uc [ S5eU$)Nz,format should not be None if len(inputs) > 0)rsuggest_memory_formatrcr`r)inputsformatrfs r: cat_compute_output_memory_format-cat..cat_compute_output_memory_formatF sgA++A.AE+++!fk...F  > !OP P r<rz3cat expects at least one tensor, but received zero!ztensor must be TensorLike, got rFrFrWcD>STRSTRSTS3$)N6Number of dimensions of tensors must match. Expected z-D tensors, but got z-D for tensor number in the listr)examplerMrsr:rUcat..y s. ' ~-A!&&J%%&C|5r<z+tensor.ndim should be 1 at this point, got c*>STRSTS3$)Nrz*-D tensors, but got 1-D for tensor number rr)r tensor_idxsr:rUr s"#LL>*!!+ L:r<c38# UHoRv M g7frK) requires_gradrMs r:rPcat.. s$Fg__gr\rrr9rrr)rZror@r'rr7rrrbrFrlrSrcrnr[rrr Exceptionrrr9canonicalize_dim validate_idxrrr)rr rrtensorrFr[filteredrrrrMrrs @@@@r:rr? si  7|qCo&*-- #B4<.!QR R WEuED$G'"1 ?vv{vv{ FFgll*5 #!* MMEH'0 F u:V\\* *{{a$A&++O LLv||A!34: ${{aN6<<?a3G$H$H OOF #91<5W=M 8} AJ " !$Fg$F!FGM ''88''     !!1!13 7C x{''- 99X # ) ) ) FF "!M "s1J J'&J'c>[SU55n[US5$)Nc3# UH7oRS:aUO URUR5S45v M9 g7fr~)rSrrrMs r:rPcolumn_stack.. s3@G1VVaZQYY 1~66s?ArW)rhrraligned_tensorss r:rr s(@GO  ""r<c[R"UR5(dU$UR(a[R "U5$[ R"U5$rK)rrr is_sparsercr;rrrs r:rr sC  ! !%++ . .  ""5)) ::e r<padc ^^ ^ ^ ^ ^[R"[T5S-S:HU4Sj5 URm [T 5m [T5S-nT U- m [R"T U:U U4Sj5 Un[ T T 5Hxm ST T - S- -mTTS:a*UR T TT*URT TT-5nTTS-S:dMOUR T SURT TTS--5nMz [ R"ST55(aUR5$[T ST 5n[ U5Ham [T5T S-S-- mT T T -TT-TTS--n[R"US:U U U UU4Sj5 URU5 Mc [R"U5n[R"UURURUR US9nUS:Xa UR[R":XaS n[R$"X5nUn [ T T 5Hwm ST T - S- -mTTS:a)U R T TTU RT TT- 5n TTS-S:dMNU R T SU RT TTS-- 5n My [&R("X5 U$) NrGrc >S[T53$)Nz1Length of pad must be even but instead it equals rY)rsr:rU!constant_pad_nd.. sCCH:Nr<c(>S[T5STS3$)Nz`Length of pad should be no more than twice the number of dimensions of the input. Pad length is z while the input has dimensions.rY)l_inprsr:rUr s 225c(;P 'r<rWc3*# UH oS:v M g7frJrS)rNps r:rP"constant_pad_nd.. s'3aE3sc F>STTT-STTSTTS-STT-S3 $)NzThe input size z, plus negative padding rrWzG resulted in a negative output size, which is invalid. Check dimension z of your input.rS)rM input_sizesl_diffrpad_idxsr:rUr sEok&1*&=%>>V7|nE#gk"2!34117! OMr<rF)rcrnrZr[rmrrRrrrorrrrrr9rrrHrr)rrrl_padc_input new_shapenew_dimrrCc_outputrMrrr rs ` @@@@@r:rr s  LL C1 N ++K  E HME U]F LL  G 65 !uqy1}% wN+NOG ""||'3'''}}[&)*I 5\c(q1uk*fqj)CL83w{;KK qL M M !//6M [[kk||))# F zekkUZZ/ ZZ &FH 65 !uqy1}% w<1 3w<!2S\!AH w{ q q!X^^A->Wq[AQ-QRH" MM($ Mr<c[R"U[R:gS5 [XS9(aU$[R"XS9$)Ncg)Nz@preserve memory format is unsupported by the contiguous operatorrSrSr<r:rUcontiguous..- sRr<r)rcrnr)rr)rsrs r:rr( s? LL...R 0O ;;q 66r<cr[R"[U5S:S5 [U6n[ US5$)Nrcg)Nz%dstack expects a non-empty TensorListrSrSr<r:rUdstack..9 +Rr<rG)rcrnrZrrrs r:rr7 1 LLW!#RS '*O  ""r<)implicitrc ^^SSKJnJnJnJn [ R RRRn[U5S:Xa&[US[5(a[US5n[ R"[U5[UR5:S5 [U5[UR5- n[!U5n [#UR5Hun mX-n X+mU"TS:H5(aTX'M$U(aOU"T5(aBU"T5(a5U"T5n U"T5n U S:XaU S:wa[ R"TS:H5 [ R"U"TS:HTT:H5UU4Sj5 [ R"TS:5 TX'M [$R&"U 5 [(R*"X [[-U[UR5U-555$)Nr)rFrGrHr~rWcg)Nz3expand: the requested shape has too few dimensions!rSrSr<r:rUexpand..Q sEr<r]c>STSTS3$)Nz3expand: attempting to expand a dimension of length  -> rrS)requested_lengthrOsr:rUr"{ sMaSPTUeTffghr<)rbrFrGrHr~rcrdrerfrgrZr@rrhrnr[rorlrvalidate_shaperrVrm)rsrr[rFrGrHr~rqrshape_ra offset_idxx_hintrequested_hintr%rOs @@r:rr> s%%--CCI  5zQ:eAh99eAh LL E c!''l"E Z#agg, &F %[FAGG$Q\  , *b0 1 1!"F %a((%&677/2!78H!IQ;>Q#6LLa( LLqAv/145h  LL)Q. /!1F Q%V   ! ! 5vs177|f'<=> r<c8URUR5$rK)rr[r[s r:rr s 88AGG r<chunksc US::aSUS3n[U5e[R"URU5nURUn[ R "XA- 5n[ R"XE- 5nXE-n[U5Vs/sHn[XX-U5PM n nUS:waU R[XXe-U55 [U 5$s snf)Nrz%Expected at least one chunk, but got r) rorrrSr[r9r:rJrmrrrh) rsr,r rlength chunk_size full_chunkstail_chunk_sizerMrs r:rr s {5fXQ?o   -C WWS\F6?+J**V01K)OBG BT UBTQfQQ^Z8BTF U! fQ[%=OP = Vs C start_dimend_dimcF[R"URU5n[R"URU5nX:XaURS:waU$[R"XUSS9up4Ub[R "XU5$[R "XU5$)Nr must_be_valid)rrrSr_collapse_view_helper collapse_viewcollapse)rsr2r3r _new_stridess r:rr s&&qvvy9I$$QVVW5G!  $99 gTI""199 >>! 00r<c[U[5(d [U[5(d [S5e[R "UR U5n[R"U5 [R"X5$)Nz!dims has to be a sequence of ints) r@rhrororcanonicalize_dimsrSvalidate_no_repeating_dimsrrev)rsrXs r:rr s\ dE " ":dD+A+A<==  " "1664 0D $$T* 99Q r<cPURS:a [S5e[US5$)NrGzInput must be >= 2-d.rWrSrjrrs r:rr %vvz233 4=r<cPURS:a [S5e[US5$)NrWzInput must be >= 1-d.rrArs r:rr rBr<rr.ct^^^[T[5(a`[R"TR 5S:H=(a [ R "TR5S5 TR5m[[T5m[R"UR 5S:S5 [R"TS:S5 [ R"URU5nURU5m[R"[T*T:*=(a TT:*UU4Sj5 TS:aTT-m[R"TTT- :*UUU4Sj5 [!UR"5nTXA'UR%X@R'5UR)5UR'U5T--5$)Nrcg)Nz'start must be an 0-dim integral Tensor.rSrSr<r:rUnarrow.. s=r<cg)Nz-narrow() cannot be applied to a 0-dim tensor.rSrSr<r:rUrF s&Ur<cg)Nz&narrow(): length must be non-negative.rSrSr<r:rUrF s&Nr<c>ST*STSTS3$)Nz0start out of range (expected to be in range of [, z ], but got )rS) dim_lengthrsr:rUrF s#BJ;-rR\Q]]hinhoopqr<c>STSTSTS3$)Nzstart (z ) + length (z) exceeds dimension size (z).rS)rLr.rsr:rUrF s'% VH4NzlZ\]r<)r@r'rcrnr rrrrr rrrSrQ _check_with IndexErrorror[rrMrS)rsr rr.rrLs `` @r:rr sO %$$ IIK1  D!7!7 !D =   e E LL1UV LL1NO   -CJ   u4*!4q  qy " LL f$$]QWW IIN <<88:q//1AHHSME4II r< norm_dimsepsch[R"URU5n[R"UR5n[ X5n[ U[5(d[S[U535e[R"XASSS9upV[R"XR-5nXF- U-nXU4$)aComputes mean and 1/std of a tensor along norm_dims. Used as a helper function for normalization layers. Args: a (Tensor): input tensor norm_dims (DimsType): dimensions to normalize over eps (float): epsilon for numerical stability Returns: out (Tensor): normalized tensor. mean (Tensor): mean of the tensor along norm_dims. rstd (Tensor): 1/std of the tensor along norm_dims. za_acc must be TensorLike, got FTr rr ) rr<rSr rr+r@r'rr7rcrr) rsrPrQra_acc biased_varrrstdrs r: _normalizerW s$'' :I33AGG< #A 9E eZ ( (=d5k]KLL~~ udJ ;;z' (D <4 C d?r< dimensionscV[U5Hn[R"X5nM U$rK)sortedrcr)rOrXr s r:_unsqueeze_multipler[ s$j! OOA #" Hr<weightbias batch_size num_channelsflattened_inner_size num_groupscx^^[R"TRS:U4Sj5 [R"UT-S:HUU4Sj5 [R"TR 5n[ TU5n SS/n [R"U UTUT-U/5n [R"U RU 5n [R"XSSS9up[R"X-5nTRRS :XaUb[R"US TUT-S /5nX-nU *U-nUb#[R"US TUT-S /5nUU-nUR5RX4/US /5nUR5RX4/US /5n[[!STR55n[#UU5n[#UU5nU U-U-nOmX- U-nUR%TR&5nS/[[!STR55-nUb[#UU5nUU-nUb[#UU5nUU-n[ UTR 5n[ U TR 5n [ UTR 5n[R("X5n [R("X5nUX4$) NrGc">STR3$)Nz=Expected at least 2 dimensions for input tensor but received rrsr:rU#native_group_norm..( sOPUPZPZ|\r<rc.>SSTRST3-$)NzDExpected number of channels in input to be divisible by num_groups, zbut got input of shape z and num_groups = rK)rrasr:rUrd, s!V #EKK=0B:, OPr<rFTrSrrW)rcrnrSrr rr+rr<rrr9r7rrrormr[r r[r)rr\r]r^r_r`rarQr input_accrTinput_reshapedrUrrVweight_reshapedwrt bias_reshapedrY unsqueeze_w unsqueeze_brunsqueeze_weightunsqueeze_biass` ` r:rr s LL a\ LLz!Q& P 33EKK@'/@AIVN]] Z!;=QRN,,^-@-@.QN~~UDJ ;;z' (D ||E!f&8-- Q LJ$>B   " EAI  !MMq*lj&@!DMM!A LLN % %z&@J ((C  0~FN&C !#u{{ 3C "4 5D "4 5D == .D == .D  r<out2normalized_shapecN^^^^SSKJn [T5n[R"US:U4Sj5 [R"TSL=(d U"TR [ T55UU4Sj5 [R"TSL=(d U"TR [ T55UU4Sj5 [R"TRU:=(a, U"TR TRU- S[ T55UU4Sj5 [R"TR5(+S5 [T5mTb [T5mTb [T5mTRU- n[[UTR55n[TX5upn Tc TbU T-n OTb TcU T-n OTb TbU T-T-n [U TR5n TRR S ;a,[U TR5n [U TR5n XU 4$) Nrsym_eqrWc >S[T5-$)NzzExpected normalized_shape to be at least 1-dimensional, i.e., containing at least one element, but got normalized_shape = )r)rpsr:rU#native_layer_norm..q sI   r<cR>S[TR5-S-[T5-$)NzQExpected weight to be of same shape as normalized_shape, but got weight of shape  and normalized_shape = rr[)rpr\sr:rUru{ s3 fll  %%    r<cR>S[TR5-S-[T5-$)NzMExpected bias to be of same shape as normalized_shape, but got bias of shape rwrx)r]rpsr:rUru s2 djj/ %%    r<cp>S[T5-S-[T5-S-[TR5-$)NzGiven normalized_shape=z, expected input with shape z, but got input of size rx)rrpsr:rUru sI)   ()    % % ekk  r<cg)Nz1native_layer_norm does not support complex inputsrSrSr<r:rUru sCr<)rmtia)rbrsrZrcrnr[rhrSr0rrormrWr+rr9r7) rrpr\r]rQrsnormalized_ndimaxisrTrrrVs ```` r:rrc s=*+O LL1  LL$G&u5E/FG  LL  Ctzz51A+BC  LL o%  KKo58 9 " #    LL    C u E F# $ :: 'D%ejj12N . s5Tr<rcg)Nz'renorm: non-positive norm not supportedrSrSr<r:rUr s Ir<cg)Nz#renorm: maxnorm must be real-valuedrSrSr<r:rUr s2Wr<c>ST3$)Nz,renorm: expected maxnorm to be >= 0 but got rS)rsr:rUr s LWIVr<rWc>STS3$)Nz/renorm: input needs at least 2 dimensions, got dimensionsrSrsr:rUr sA${Sr<T)r rr gHz>r)rcrnr@rrSrrrormr rlinalg vector_normrrrr) rr r rracc_typercrQ norm_factorrSs ` @r:r1r1 s_  LLZ7++-TU LLQIJ LL w ((*W LL1 V ::D LL qS  s +CuT{#K**5;;7H;;||'' k4x( ||''+t'L C++dWng.DcJK;;00kkJ   + + --r<n_fft hop_length win_lengthwindowcenterpad_mode normalizedonesidedreturn_complexalign_to_windowc ^^^^^^[R"TSL=(d TRTR:HUU4Sj5 [R"U(+=(d U SLS5 UbUOTS-mUbUOTmU c[TR5=(d+ TSL=(a [R "TR 5n [R"U S5 OU n [R"[R"TR 5=(d [R "TR 5S5 [R"STRs=:*=(a S:*Os S5 TRn U S:XaTRS 5mU(aS TR- n TS-n/[R"SU 5QTRQn[RTRU5X/U5mTRTR!5U S5mTR!S5m[R"S Ts=:=(a T:*Os UU4S j5 [R"TS :U4S j5 [R"S Ts=:=(a T:*Os U4S j5 [R"TSL=(d TRT4:HUU4Sj5 TT:aSTc*[R""TTR TRS9mTT- S-n[R%TUTT- U- /5mU(d(U (a!TT- S-n[RTUU/U5mTR'STTS9mTbTT-m[R "TR 5nUbUOU(+nU(aSOSnU(a=[R"U(+S5 [R(R+TSUS9nO[R(R)TSUS9nUR-SS5 U S:XaUR/S 5nU (aU$[R0"U5$)NcB>STR3STR3-$)NzAstft input and window must be on the same device but got self on  and window on r8rrsr:rUstft.. s'OPUP\P\~ ^ / 0r<cg)Nz6stft only supports align_to_window for center = False.rSrSr<r:rUrsHr<cg)Nzstft requires the return_complex parameter be given for real inputs, and will further require that return_complex=True in a future PyTorch release.rSrSr<r:rUrscr<cg)Nz:stft expected a tensor of floating point or complex valuesrSrSr<r:rUrsLr<rWrGcg)Nzstft expected a 1D or 2D tensorrSrSr<r:rUrs/Pr<rrc>STST3$)Nzstft expected 0 < n_fft <= z, but got n_fft=rS)r.rsr:rUr+s-fX5EeWMr<c>ST3$)Nz0stft expected hop_length > 0 but got hop_length=rS) hop_length_sr:rUr/s B;-Pr<c>ST3$)Nz9stft expected 0 < win_length <= n_fft but got win_length=rS) win_length_sr:rUr3s KK=Yr<c0>STS3STR3-$)Nz8expected a 1D window tensor of size equal to win_length=rJzbut got window with size rK)rrsr:rUr7s#F{mSU V)&,,8 9r<rr]) dimensionrQsteporthocg)Nz9Cannot have onesided output if window or input is complexrSrSr<r:rUrROr<r rc)rcrnr9r0rrrrrSr itertoolsrr[rrr rQr&rrfftrfft transpose_squeeze_ view_as_real)rrrrrrrrrrrreturn_complex_ original_ndim extra_dims pad_amountextended_shapeleftinput_pad_amount complex_fftrcrrr.rs`` ` @@@r:r2r2 s LL$7&--5<<7  LL  -o-H !+ 6*EQJK * 6*EK**, $  G5#9#9&,,#G     ) LL U[[)PU-C-CEKK-PL LLejj%%A%'PQJJM" ^ aZ I9++Az:IU[[IN3j5MxX 5::< 45 ZZ]F LL EVM LLaP LL K  5 Y LL$8&,,;.8 U >ZZ 5;;u||TF #)%%ftU[5H45O.PQ o!K/A5!13C DhO LL2E L DE ((5K#/x_H 7dD O O iinnUn6iimmErm5NN1all1o!3>u'9'9#'>>r<c  ^^^[R"TSL=(d TRTR:HUU4Sj5 UbUOUS-n UbUOUn [R"[R"TR 5UU4Sj5 TR S5n TR S5mXU S- --n [R"TR5S:S5 [R"S TRs=:*=(a S :*Os U4S j5 UbUOTU:gnU(a%[R"US -S-T:HU4S j5 O[R"UT:HU4S j5 [R"SU s=:=(a U :*Os S5 [R"SU s=:=(a U:*Os S5 [R"TSL=(d TRU 4:HS5 Tc@[R"TR 5n[R"XTRS9nOTnX:wa%X- S -n[RUUX- U- 4S5nTRnTRS :XaTRS5mTRSS 5mU(aSOSnU (a=[R"U(+S5 [R R#TSUS9mO[R"TSL=(d% [R"TR 5(+S5 U(dTR%SSUS -S-S9m[R R'TSUS9mTR S 5U:wa [)STR S 5SU35eTUR+SSU/5-n[R-UUR S5U 4SUU S9n[R-UR/S 5R1SX45UR S5U 4SUU S9nU UR S5:wa![)SU SUR S5S35eU UR S5:wa![)SU SUR S5S35eU(aUS -OSnUbUU-nOU(aXS -- nOU n[3SUU- 5nUR%SUUS9nUR%SUUS9nUU- nUS :XaUR5S5nUU :a1[6R8"SS S9 [RUSUU - 4S5nU$) NcB>STR3STR3-$NzBistft input and window must be on the same device but got self on rr8rsr:rUistft..ps(PQVQ]Q]P^ _ / 0r<rc@>STRSTR3-$rr8rsr:rUr{s! Pofmm_= >r<r]rWrcg)Nz"istft input tensor cannot be emptyrSrSr<r:rUrs,Pr<rGrc">STR3$)Nz8istft expected a tensor with 2 or 3 dimensions, but got rrsr:rUrsJ5::,Wr<c>SST3-$)NMistft expected the frequency dimension (3rd to the last) of the input tensor z3to match n_fft / 2 + 1 when onesided=True, but got rSfft_sizesr:rUrs_GzRSr<c>SST3-4$)Nrz,to match n_fft when onesided=False, but got rSrsr:rUrs_@ KLr<cg)Nz+istft expected 0 < hop_length <= win_lengthrSrSr<r:rUrrIr<cg)Nz&istft expected 0 < win_length <= n_fftrSrSr<r:rUrs*Rr<cg)NzDInvalid window shape. window has to be 1D and length of `win_length`rSrSr<r:rUrsVr<rrcg)Nz9cannot have onesided output if window or input is complexrSrSr<r:rUrrr<rcg)Nz:Complex windows are incompatible with return_complex=FalserSrSr<r:rUrPr<)r rr.z%Expected input.size(2) == n_fft, got z != )rr rQrzexpected_output_signal_len (z) != y.size(1) (rKz) != window_envelop.size(1) (zThe length of signal is shorter than the length parameter. Result is being padded with zeros in the tail. Please check your center and hop_length settings stacklevel)rcrnr9rrrrQrrSr[corresponding_real_dtyper&rrrrrifftrirfftrr unfold_backwardrrrkrwarningswarn)rrrrrrrrr.rrrn_framesexpected_output_signal_len onesided_ real_dtypewindow_rrrcy_tmprwindow_enveloprendrs` ` @r:r3r3as LL$7&--5<<7 !+ 6*EQJK * 6*EK LL u{{+ zz"~Hzz"~H!&1 )E!E LL"$PQ LL UZZ1W%0h%6GI QJNh &    X    LL K&&;&= LL K  5 "R LL$8&,,;.8V ~33EKK@ **[5<<P#)&&wu7JT7Q0RTUVJJM zzQ" OOAq !E 7dD M O  u"48 dN F%"8"8"FF P LLRq!aLHE 2D9 zz!}3EJJqM?$ug N   GLL!Q/ /E  ZZ]$>?    A)) Aq(23ZZ]$>?   *N"QVVAY.*+E*FFVWXW]W]^_W`Vaab c  "^%8%8%;;*+E*FFcdrdwdwxydzc{{| }  !EQJaE fn (A:5( C%K F QeF3A#**qf*MN NA IIaL ''  `  Q.H(H$I1 M Hr<a_shapea_striderrc^^^ ^ [U5n[R"XRSS9m US:XaSOUT m US:XaSOUT n[R"TT :*U U U4Sj5 [R"TS:U4Sj5 [ U5n[ U5nUR T5 UR U5 T U:aUT T- T-S-UT 'UT ==T-ss'Xx4$)NT) wrap_scalarrrWc>STSTST3$)Nz%Maximum size for tensor at dimension z is z but size is rS)r max_sizerQsr:rU*_get_unfold_shape_stride.. s7uD -X\W]^r<c>STS3$)NzStep is z but must be > 0rSrsr:rUrs(4& 01r<)rZrrrcrnror) rrrrQra_ndim last_strider[stridesr rs `` @@r:_get_unfold_shape_striders\F   ECaKqWS\H{! K LL ^  LL q1 ME8nG LL NN; V|Cj4'D014c   >r<c [R"USS9n[R"[ U5[ UR 5:S5 [ U5S:Xa[R "U5$[ U5UR- nS/U-nUR HnURU5 M [S[X1555nSU;aJ[R"UURURUR[R"U5S9$Un[R "U5n[#U5Hup[%XgX['US55upgM [)[#U55n U R+[,R."S5SS 9 [U 6upUR1U5n [R "U 5n U R3U 5nUR5U5$) NFrwcg)Nzbrepeat: Number of dimensions of repeat dims can not be smaller than number of dimensions of tensorrSrSr<r:rUrepeat.."str<rrWc3.# UH upX-v M g7frKrS)rN padded_size repeat_sizes r:rPrepeat..-s(G $K !(GrT)keyreverse)rrrcrnrZr[rrSrrhrrrr9rrmake_contiguous_strides_forrlrrkrosortr itemgetterrrr)rs repeat_shapenum_new_dimensions padded_shapedim_size target_shapeurtensor_shapeurtensor_strider enumerated_stride permute_order_sorted_striderepeat_xtensor cloned_resultpermuted_results r:rrs33L5QL LL LS\)t  <A{{1~\*QVV33++LGGH%(+L(GL  L{{ ''88//55a8   "N77 EO"<0 *B SC!STSTRT3$)Nza.size(z) expected to be 1 but got rK)rrasr:rU/_reshape_view_helper_core_alg..sgcU"=bhhsm_Mr<)rSrr split_dimr[r7rrMrorrcrnrr) rsr[r r.last_dimaccumrrr:rrras @@r:_reshape_view_helper_core_algrVs( C B "''>{$=fX\Rww{HXrxx/ABB'C  RXXc] "'C  n! 1HC RXXc] "En! #: ',&A&ACD' #I  ==228 qxxzlZrsxryyz{ o%S#&B ?S&1BAg_d - HHSMQ  M R  - Qw}}Q r<c6^^[R"TSS9m[R"TTR55mTR5S:Xa![ TT[R "T55$TR S:XaHTnTH#nUS:wa[SU35e[US5nM% UTLa[R"T5$U$[T5S:XaRTnTRH#nUS:wa[SU35e[US5nM% UTLa[R"T5$U$[T5(a[T5S:Xa8TR S:a([R"TTR5/S/5$[T5S:Xa5TR S:Xa%TSnTSn[R"TXV/US/5$[!["R$TS5n[R&"TR5U:HUU4S j5 [)TTU5$) NFrwrrWzSTRSTS3$)Nz&Could not reshape a tensor with shape r rrKr{sr:rU&_reshape_view_helper..s8 AYZ_Y``abr<)rr infer_sizerrrrSrrrrrZr[rrrcrrrrnr)rsr r[_ar.dim0dim1 shape_numels` ` r:_reshape_view_helperrs  , ,UU CE   UAGGI .E wwyA~!UE$E$Ee$LMM vv{ F{$RSYRZ[2r"B  7==# #I 5zQ ggF{$UV\U]^RB  7==# #Ia  u:?qvvz##A {QC8 8 u:?qvv{8D8D##A|dAY? ?ua0K LL  [ b )E: >>r<r[c [U/UQ7SS06$)Nr T)rr{s r:rrs  ;E ;d ;;r<c@URUR55$rK)rrQrr6s r:rrs << %%r<shiftsc[R"URU5n[U[5(dU4n[U[5(dU4nUR 5S:XaUR 5$UR5S:Xa![U5S:a[SUSS35e[U5n[U5nUS:wdUS:waUS:Xa [S5eUS:XaKUS:XaE[R"[R"U5US5RUR5$X4:wa[SUSU35eUS::a[!SU35eUSS nUSS n[R"XS4US5n[R"XuU5$USnURUn XS- U -n [R""XR$S 9n UR'U[R("X-U 55$) z/Reference implementation of :func:`torch.roll`.rzDimension specified as z but tensor has no dimensionsrWz`shifts` requiredz*shifts and dimensions must align. shifts: z, dims: zExpected len_dims > 1, got Nr8)rr<rSr@rrrr rZrOrjrcrrr r[rrr9rPr) rsr rX len_shiftslen_dims tail_shifts tail_dimsfirst_dim_rolledr rQrras r:rrs  " "1664 0D fh ' ' dH % %w wwyA~wwy uuw!|D A %d1gY.K L  VJ4yHQ(a- ?23 3 q=Z1_::emmA.:??H H  !xj!IJ JQRj H !::a)tAw?zz*CC q'C 773>#uzz%+t< ==r<rrWkc[U5S:wa[S[U535eURS:a[SUR35e[R"URU5nUSUS:Xa[SUSSUS35eUS-nUS:Xa6[ R "[ R"XS45USUS5$US:Xa[ R"X5$US :Xa6[ R "[ R"XS45USUS5$UR[ RS 9$) z0Reference implementation of :func:`torch.rot90`.rGz2expected total rotation dims == 2, but got dims = z/expected total dims >= 2, but got total dims = rrWz7expected rotation dims to be different, but got dim0 = z and dim1 = rrr) rZrjrSrr<rcrrrr`)rsr(rXs r:rr0s9  4yA~@T L   vvzLQVVHUVV  " "1664 0D Aw$q'Ed1gYl[_`a[bZc d   AAAvuzz!1gZ8$q'47KK azz!"" auzz!1gZ8$q'47KKwwU%<%STS3$)Nz4stack expects each tensor to be equal size, but got z at entry 0 rS) entry_shapesr:rU%_check_stack_inputs..WsJ;-Wcdr<)rbrsr[rmrZrcrn)rrsrMr,s @r:_check_stack_inputsr.PsK<!*""K 1c'l # 7:##[ 1 d $r<c[U5S:Xa [S5e[R"USRS-U5nX SR:ae[ U5 [ USR5nURU[U55 [R"X5nURU5$[R"UVs/sHoURU5PM snU5$s snf)Nrz$stack expects a non-empty TensorListrW) rZrrrrSr.ror[insertrcrr r)rr  wrapped_dim result_sizesrrs r:rr[s 7|qCDD((1).rr<rW)rcrnrZrrSrrs r:rrsQ LLW!#RS '*Oq!#?A&&  ""r<cr[R"[U5S:S5 [U6n[ US5$)Nrcg)Nz%vstack expects a non-empty TensorListrSrSr<r:rUvstack..rr<)rcrnrZrrrs r:rrrr<rVc*[R"URU5n[R"[ U5S:gS5 UR [URSU5[U5-[URUS-S5-5$)Nrcg)Nz"unflatten: sizes must be non-emptyrSrSr<r:rUunflatten..s*Nr<rW) rrrSrcrnrZr rhr[)rsr rVs r:rrso   -C LLUq"NO 66% &u5aggcAgi>P8QQ RRr<rc8^[R"URT5m[R"[ UR 5S:S5 UR TS:Xag[U4Sj[R"XR TT555$)Nrcg)Nz5Dimension specified as 0 but tensor has no dimensionsrSrSr<r:rUunbind..sGr<rSc3R># UHn[R"UT5v M g7frK)rcr)rNsr s r:rPunbind..s# +SaEMM!S ! !+Ss$') rrrSrc _check_indexrZr[rhr)rr s `r:rrs~   -C  AGG qG wws|q +0+=+=as+S   r<r;rc\UR[RS9RXU5$r)rrcr`rO)rOr r;rs r:rNrNs+ 77!8!87 9 E E F r<c^[R"URU5n[R"TRS:*U4Sj5 URS:XaUR S5OUn[ S54U-T4-nX4U'U$)NrWc$>STRS3$Nz(Index should have dimension 1 or 0 (got rKrr;sr:rUindex_copy_..:5::,aHr<r)rr<rSrcrnrslice)rOr r;rrras ` r:rOrOsq  ! !!&&# .C LL aH &&A+ A1A ;.3 % )C cF Hr<c[XX#SS9$)NFinplace _index_fillrOr r;rs r:rQrQs quU ;;r<c[XX#SS9$)NTrOrQrSs r:rRrRs quT ::r<rPc^^[R"TRS:*U4Sj5 [T[5(a)[R"TRS:HU4Sj5 O5[R "TUR URURS9mURS:HnU(aURS5OUn[UR5nTR5Xq'TRU5mU(a[RO[R nU"XaTT5n U(aU$U(aU R#S5R%5n U R'5UR'5:wa?[(R*"U5(a$[,R."XR'55n U $)NrWc$>STRS3$rIrrJsr:rU_index_fill..rLr<rc$>STRS3$)NzSTRS3$rIrrJsr:rUindex_select..rLr<rr) rr<rSrcrnrrrNrrMrr)rOr r;ras ` r:rPrPs  ! !!&&# .C LL aH zzQ"vv{"--a E8JKK ;.3 % )C 6  5+F+Fq+I  JJr<c^^SSKJm UcY[S[TR555nU(a[ R "TU5$[ R"T5$TRn[R"X15n[U[5(aU4OUnUS:Xa9[U5S:XdUS:Xd[SU35e[ R"T5$[UU4SjU55n[U5S:Xa[ R"T5$[U5S:Xa[ R "TU5$[U5n[!USS 9nUHn[ TU5mM T$) Nrrc3:# UHupUS:XdM Uv M g7fr~rS)rNrarQs r:rPsqueeze..-sJ*# UH&nT"TRUS:H5(dM"Uv M( g7fr~rK)rNdrsrFs r:rPrc>s&DDqN1771:?$CDs!1 1rWT)r)rbrFrhrlr[rrrrSrr<r@rrZrrorZ)rsr rXrS dims_listrMrFs` @r:rr(s*D {J)AGG*<JJ)-u}}Q%C5==3CC 66D  ! !$ ,CS))C6sD qyD Q$$, J4&Q }}Q DDD DD 4yA~}}Q 4yA~}}Q%%T Iy$/I  AqM Hr< split_sizesc^^^[[T55H!n[R"TUS:S5 M# [R"[ [ R"T5TRT:HUUU4Sj5 /nTR5nTHen[TR5nXgT'URTRUTR5U55 UTR5TU--nMg U$)Nrcg)NzCsplit_with_sizes expects split_sizes have only non-negative entriesrSrSr<r:rU"split_with_sizes..SsYr<cV>S[R"T5STRT3$)NzSplit sizes add up to z but got the tensor's size of )rRrr[)r rrgsr:rUrjXs,(k)B(CCabfblblmpbqarsr<)rmrZrcrnrNrorRrr[rSrorrrM)rrgr rMsplitsr split_sizers``` r:rrJs 3{# $ Na  Y %   [!TZZ_4s F  "F! $ ##  dooiGH$++-,z99 " Mr<indices_or_sectionsc[R"URU5nURS:Xa Sn[U5e[ U[ 5(alUR RS:waSUR 3n[U5eUR[R:waSUR3n[U5e[ U[5(d%[ U[ 5(aURS:Xa[ U[5(aUOUR5nUS::aSU3n[U5eURUn[R "Xe- 5nXe-n/n [#U5H n X:aUS-OUn U R%U 5 M" ['[(R+X US95$Un [ U[ 5(a;URS:waS URS 3n[U5eUR-5n S/[/U 5-URU/-n [#[1U 5S- 5V s/sHoU S-X- PM n n ['[(R+X US95$s sn f) NrzXtensor_split: received a rank zero tensor, but expected a tensor of rank one or greater!rz\tensor_split: if indices_or_sections is a tensor it must be on the CPU, but received one on zgtensor_split: if indices_or_sections is a tensor it must have long dtype, but received one with dtype zAtensor_split: number of sections must be greater than 0, but was rWr zntensor_split: non-scalar indices_or_sections tensors must have only one dimension, but received a tensor with r)rrrSror@r'r9r7rrclongrr!rr[r9rJrmrrhrrtolistrorZ)rsrnr _dimrsectionsrmin_split_sizenum_splits_one_extrarg split_idxrmindicesrMs r:rrjs|  ! !!&&# .Dvv{ho%z22  % % * *e 3'':'A'A&BD S/ !  $ $ 200C0I0I/JL S/ !%w//& 338K8P8PTU8U-v66 $))+  q=UV^U_`CS/ !774=H$78'2 xI4"#    z * )T**1t*DEE& ): 6 6"''1,22E2J2J1K;X!o%)002G#W %7S[TR5-S-$)NzPtorch.hsplit requires a tensor with at least 1 dimension, but got a tensor with dimensions!rrSrsr:rUhsplit.. ^!&&k  r<rc|>S[T5-S-[TRT5-S-[T5-S-$)Nz0torch.hsplit attempted to split along dimension z , but the size of the dimension $ is not divisible by the split_size rrx)rsr rmsr:rUr}sSBc(45aggcl#$9 9 j/ "  r<c >S[T53$)Nzhsplit(): received an invalid combination of arguments. Expected indices_or_sections to be of type int, list of ints or tuple of ints but got type r6rnsr:rUr}  !456 8r< rcrnrSr@rr[r _check_typerorh)rsrnrgr rms`` @@r:rrs LL !  vv{!C%w//(  1_ ? !:a!?  Az3// &u 6 &K ; ,,r<c^^^[R"TRS:U4Sj5 [T[5(aJTm[R"TS:g=(a TR ST-S:HUU4Sj5 [ TTS5$[R"[T[[45U4Sj5 Tn[ TUS5$)NrGc:>S[TR5-S-$)NzPtorch.vsplit requires a tensor with at least 2 dimension, but got a tensor with r{r|rsr:rUvsplit..r~r<rc0>STRSSTS3$)NzQtorch.vsplit attempted to split along dimension 0, but the size of the dimension rrrrK)rsrmsr:rUrs%3771:,6, r<c >S[T53$)Nzvsplit(): received an invalid combination of arguments. Expected indices_or_sections to be of type int, list of ints or tuple of ints but got type r6rsr:rUrrr<r)rsrnrgrms`` @r:r r s LL !  %w//(  1_ =j!8A!=  Az1-- &u 6 &K ; **r<rc^UR5m[R"TS;U4Sj5 TS:Xa[R"X5$[R"X5$)N)rWrGc>STS3$)Nz'diag(): Supports 1D or 2D tensors. Got DrSrsr:rUdiag..s"I$q Qr<rW)r rcrnrr)rrrSs @r:rrsQ 88:D LL Q qy--""400r<rdim2c `^^SSKJnJn [R"U5nUR X#U5m[ R"TRTR:H=(a5 U"U"S[TRTR5565(+UU4Sj5 [TT5 U$)Nr)rFr~c3.# UH upX:gv M g7frKrS)rNd1d2s r:rP#diagonal_scatter..(sG,F&"RX,Frc<>STRSTR3$)NzCexpected src to have a size equal to the diagonal of the input.Got z for a diagonal of shape rK)rrsr:rU"diagonal_scatter..*s"yyk24::,@r<) rbrFr~rclone_preserve_stridesrrcrnrSrr[r) rrrrrrFr~rrs ` @r:rrsM  & &u -C <<d +D LL SXX  GC CII,FG H   @  D# Jr<c^^UR5n[R"TUS9m[R"TUS9m[R"TT:gUU4Sj5 UR 5nUS:aT[ [UR5TUR5TU- 5S5nXQUR5T-- nOS[ [UR5TU-UR5T5S5nXQUR5T--n[UR55VVs/sHupxUTT4;dMUPM n nnU RU5 [UR55VVs/sHupxUTT4;dMUPM n nnU RUR5TUR5T-5 URXUS9n U $s snnfs snnf)z, Reference implementation of torch.diagonal )rarankc>STST3$Nz(diagonal dimensions cannot be identical rJrSrrsr:rUdiagonal..@ HbQUPVWr<r)rQrMrS) r rrrcrnrSrrrQrMrlrr) rrrrnum_dimsrS diag_sizerMrCrVrrs `` r:rr1sxxzH  ! !d :D  ! !d :D LL  W((*N{GDIIK$5tyy{47H67QRTUV 4;;=#666GDIIK$5$> D@QRTUV 4;;=#666$TYY[1 K141QtTl5JQ1E K LL&t{{}5O5TQ4,9Nq5GO NN4;;=&t)<<= __%_ WF M LPs5G G =G&G&c*^^URS-n[R"UTS9m[R"UTS9mTT:aTTsmmU*n[R"TT:gUU4Sj5 UR S5nUS:wa[ UR5n[R"U5US'[R"X`RURSS9nUS:aXp4OX4n[R"USS9nU[R"U5- nURT5RST5n[R "XPR[R"S 9n [R "XU-UR[R"S 9n XRS5:Hn [%['UR55V s/sHoTT4;aUOSPM n n U R)U 5n [R*"X5R-5$s sn f) z. Reference implementation of torch.diag_embed rW)rrac>STST3$rrSrsr:rUdiag_embed..orr<r]rFrr9rrpr9r)rSrrrcrnrQror[rRr2r+rr9rrrrrsrmrZr mask_tensorr)rrrrrrt_shapezpaira_rangeb_rangecondrM cond_shapes `` r:rrXs 66A:D  ! !t 6D  ! !t 6D d{4 d LL  W vvbzH {qww-ll6* KKwwqxxu U!v! IId #HLL(( D!!"d+All8HHEKKHGll6!!((%++G ''+ +D@Ec!''l@ST@S1D$</(Q6@SJT << #D   T % 0 0 22 UsHc^^ ^ UVs/sH*oR5S::aURSS5OUPM, nn[R"SU55nUSRm /nSn[ U5Hum m[ R"TR5S:HU U4Sj5 [ R"TRT :HU U U4Sj5 TRupg[ R"Xe4T TRS9n[ R"XcU- U- 4T TRS9n U[ R"UTU 4SS 9/- nXW- nM [ R"USS 9$s snf) z. Reference implementation of torch.block_diag rWr]c3># UHoRSv M g7fr~rK)rNrs r:rP'_block_diag_iterable..sBzVazsrrGc2>STSTR5S3$)Nz5Input tensors must have 2 or fewer dimensions. Input z has rrp)rMrsr:rU&_block_diag_iterable..s CuVZZ\N+7r<c0>STSTSTRS3$)NzCInput tensors must all be on the same device. Input 0 is on device z and input z is on device rr8)r9rMrsr:rUrs&$$*8;qc VWYr<rrp) r r rRrr9rlrcrnr[r+rr) rr tensors_2dncolsr col_startrowcolrrightr9rMs ` @@r:_block_diag_iterablersLKRJQjjla/ ArV;' LLBzB BE ] ! !F FIz* 6 JJLA  7  MMV # Y <<{{C+F&,,O )#c) *6  599dFE2:;; #+& 99V ##;s1E"c[U5$)z This is used as an input to PythonRefInfo. `torch.block_diag` expects arguments splatted, but `aten.block_diag` expects only one argument that is a list of Tensors. )rrs r:rrs  ((r<rtcURS:a[SURS35e[U[5(a>US:XdURSU-S:wa"[SSURSSUS 3-5e[ XS5$) NrzPtorch.dsplit requires a tensor with at least 3 dimension, but got a tensor with r{rrGz3torch.dsplit attempted to split along dimension 2, zbut the size of the dimension rr)rSrjr@rr[r)rsrts r:rrsvvz^_`_e_e^ffr s  (G$$(a-1771:;PTU;U A.qwwqzl:^_g^hhij k   Q ''r<cURS:a[SURS35e[R"USURS:aS5$S5$)NrGz7t() expects a tensor with <= 2 dimensions, but self is rrrW)rSrjrcrrs r:rrsS vvzEaffXQ O   ??1aaffqj 88a 88r<cl[R"URS;S5 UR5$)N)rrGcg)NzcThe use of `x.T` on tensors of dimension other than 0 or 2 to reverse their shape is not supported.rSrSr<r:rUT..s 7r<)rcrnrSrrs r:rrs, LL &  335Lr<c.[R"U5$rK)rrrs r:rrrr<rc*[R"URX45up4URS::dX:Xa[RR U5$[ [UR55nXEU'X5U'[R"X5$r0) rr<rSrrdefaultrormrcr)rsrr_dim0_dim1rs r:rrsq**166D<@LEvv{dlzz!!!$$aff &L == ))r<ct[URUR5XU5upEURXE5$rK)rr[rMr)rrrQrr[rs r:rrs4. DKKM9DNE ??5 **r<c\URXU5R[RS9$r)rrrcr`)rrrQrs r:rrs0 ;;y - 3 3-- 4 r<cURn[R"Xc5nUS:XaU"URS5SXES9$URUS-5n[R "UR UURS9nURS5U:*n[Xc- S- 5Hn URS5nM [R"XU5n U"XXES9$)Nr)r rrrWr8r]) rSrrrrcrr[r9rmr) rinitrsr rrrSrgrrmasked_as r:_cumsumprod_commonrs 66D   +C qyAKKN@@ C!GA aggcl188 4B <<?b D 4:> "~~b!#{{4D)H  88r<c "[[SXX#S9$)Nrrrrsr rr)rrrsr rrs r:rr/s 3Q!E SSr<c "[[SXX#S9$)NrWr)rrrs r:rr:s 4a1U TTr<cxURS-n[R"X!5n[R"X4US9$)NrWr)rSrrr expand_dims)rsr rSs r:rrFs6 66A:D   +C   QT 22r<c$SSKJnJn [R"[ [ [[ -U5SS9n[RRRR(dU"UU5(aU"X5$[U/UQ7SS06$)Nr)_view_has_unbacked_input_view_unbacked_metaFrwr )torch._subclasses.fake_implsrrrrr r%rhrcrdrerfrgr)rsr[rr shape_tuples r:r r Ss22 Yy) )51EK xx$$::>V ??#122  >STRSTRS3$)Nz]torch.take_along_dim(): input and indices should have the same number of dimensions, but got z dimensions for input, and z dimensions for indicesr)rsrxsr:rU take_along_dim..zs% --.VVH4O||n3 5r<c$>STRS3$)Nz?torch.take_along_dim(): dtype of indices should be int but got z insteadr)rxsr:rUrs M}}oX 'r<r]r) rcrnrSrrrgatherr ror[rQinfer_size_shapesr) rsrxr  self_sizesbroadcast_shapeindices_broadcast indices_sizesself_broadcastrs `` r:rrts' LL ',,  LL w}}-  {||AFF2J7<<+;<<!''] !,,s+ 11*gllnM(/BW]]+ VVC[ 11-J%a9"&&s+-8||N1BCCr<rr%r9rr&rrc ^[R"T[R:gS5 [R"U5nT[R :Xa[R "U5nOmT[R:Xa[R"U5nOB[R"T[R:HU4Sj5 [R"U5n[R"UUUUUUUS9$)Ncg)Nz8torch.empty: the Preserve memory format is not supportedrSrSr<r:rUempty..sJr<c>STS3$)Nz/torch.empty: received an unknown memory format rrSrsr:rUrsEm_TUVr<rr%r9r&r) rcrnr)rrr`rchannels_last_3d!make_channels_last_3d_strides_for channels_last!make_channels_last_2d_strides_forr)rr%r9rr&rr[rs ` r:rrs LL...J  , ,U 3E///33E: %00 099%@ U00 0 V 99%@    # r<c2[R"UUUUUS9$)Nr)rr)r[physical_layoutrr%r9rr&s r:rrs&    #  r<rr%r9r&cUc UROUnUc UROUnUc UROUn[R"UUUUUS9$)Nrr9r&r%)rr%r9rcr)rsrQrr%r9r&s r:r!r!sQ}AGG%EQXXVFQXXVF ;;   r<c Uc UROUnUc UROUnUc UROUn[R"UUUUUUS9$)z< Reference implementation of torch.Tensor.new_empty_strided r)rr%r9rcr)rsrQrMrr%r9r&s r:r"r"sV }AGG%EQXXVFQXXVF      r<rc [R"U5nUc[R"5n[R"UU[R :XaSOSUUUUUS9$NFrrrrrcrrrrr%r9r&rrQs r:r+r+sZ  + +D 1D }'') :: %**$!# r<c Uc UROUnUc UROUnUc UROUn[R"UU=(d UR[R :XaSOSUUUUUS9$rrr%r9rcrrrsrQrr%r9r&rs r:r%r%)sm}AGG%EQXXVFQXXVF :: "177uzz1q# r<c [R"U5nUc[R"5n[R"UU[R :XaSOSUUUUUS9$NTrWrrrs r:r&r&DsZ  + +D 1D }'') :: ## r<c Uc UROUnUc UROUnUc UROUn[R"UU=(d UR[R :XaSOSUUUUUS9$rrrs r:r$r$^sm}AGG%EQXXVFQXXVF :: !!''ejj0a# r< fill_valuec Uc UROUnUc UROUnUc UROUn[R"UUUUUUS9$)Nr)rr%r9rcr)rsrQrrr%r9r&s r:r#r#ysT}AGG%EQXXVFQXXVF ::   r<cU$rKrS)rQrrs r: empty_outrs Jr<)rr9r%r&rrc PUc UROUnUc UROUnUc UROUnU[R:wa$[R "UR UUUUUUS9$[R"U5upx[R"UR UUUUUUS9$)Nrr) rr%r9rcr)rr[r3compute_elementwise_output_logical_to_physical_permr) rsrr9r%r&rrlogical_to_physical_permrs r:rrs}AGG%EQXXVFQXXVF---{{ GG'!'   AA!D      # r<rc 6^^^[R"U5 [R"U5 [R"[R "U55n[ T[5(a [S5e[ T[5(a [S5e[ T[5(a [S5eTcTmSm[R"TS:gS5 TS:a[R"TT:S5 O!TS:a[R"TT:*S5 Sn[R"U"T5=(a U"T5UU4S j5 [R"U"T5U4S j5 TTT4n [R"S U 55n Uc,U (a[RO[R"5n[R"U5n U (dU (a![T5n [T5n [T5nU[R:XdU (a,[!WS:5[!US:5- nW W - U-U- U-nO["R$"TT- T- 5nU (a[&R("UW WUUUS 9$[&R("USS [RUSS 9nU (a[R*O[R,"X55n[/UU5nTTU--n[/UU5nU(aUR1S5 U$)Nz%arange does not support complex startz#arange does not support complex endz$arange does not support complex steprcg)Nzstep must be nonzerorSrSr<r:rUarange..s$:r<cgNz7upper bound and lower bound inconsistent with step signrSrSr<r:rUrMr<cgr rSrSr<r:rUrr r<cf[U[5(+=(d [R"U5$rK)r@rr9rSrTs r: is_finitearange..is_finites!a!566J$--:JJr<c>STST3$)Nzunsupported range: r$rS)rrsr:rUrs%eWD6r<c>ST3$)Nzstep must be finite but got rSrsr:rUrs .tf5r<c3B# UHn[U[5v M g7frKrLrNrs r:rParange..sIDS 3 8 8D)rrrr9rrWFT)r check_layoutcheck_pin_memoryrcr9device_or_defaultr@rrrnrRrrsrrrrr9r:riotarq get_acc_typer+requires_grad_)rrrrr%r9r&rrr integer_args is_integerxstartxendxsteprlr.r;rrs``` r:rrs v :& \\%11&9 :F%!!DEE#wBCC$  CDD { LL:; ax 5L M   5L M K LL%+Ys^6 LL$5 3 D<<IDIIL }+ 1H1H1J''.J\s|   |519oUQY/-%'#-%7C%K4/0zz '    JJ kk  E# (:(:5(I $E+< =E TE\ !F $VU 3Fd# Mr<)rrr\c X/n[U[5(aURSU5nOURU5 [U[5(d[ S[ U535eUR5S:n[R"XBS- U5n[R"XAU5nXQU- -U-n[R"[U66nUR5U:wa[R"Xx5n[!X75$)NrSz)weight must be Tensor at this point, got rrW)r@r!r#rrhrr7r2rcrr"compute_elementwise_output_stridesrrMrrZrD) rrr\rrcoeffbaserCrMs r:rZrZ/s\F&&!!F+ f ff % %HfWXX ::<3 D KKqj& 1E ;;t% (D Ek "T )F  5 57G7P QF }}& ##F3 ' 77r<)rr9r%r&rstepsc ^^^^^[T[5(a>[R"TR 5S:HS5 [ T[ U55m[T[5(a>[R"TR 5S:HS5 [ T[ U55m[R"STTT455(a`[R"[R"55mTcTmOO[R"[R"T5UU4Sj5 OT=(d [R"5m[T[R5(d[S[T535e[R "[T["5UUU4Sj5 [T["5(d[S[T535e[R"TS:S 5 UUUUS .nTS:Xa[R$"SS T0UD6$TS :Xa~[T[5(aN[R&"T44S T0UD6n [R(R*R,R/U T5$[R$"T4T4S T0UD6$[R0"ST40UD6n [R2"T5(d[R4"T5(a[R6OTn [R8"U [:R<U 5up[?[ U S 9nTT- TS - - n[R@"U TS- :TX"U 5--TX"TS - U - 5-- 5n[ UT5$)NrcgNz:linspace only supports 0-dimensional start and end tensorsrSrSr<r:rUlinspace.._rr<cgr(rSrSr<r:rUr)err<c3B# UHn[U[5v M g7frKr@rrs r:rPlinspace..iL8KJsG,,8Krc>STST3$)Nzlinspace(): inferred dtype z& can't be safely cast to passed dtype rS)default_complex_dtypersr:rUr)rs56K5LLrsxryzr<dtype must be torch.dtype, got c>S[T5RS[T5RS[T5RS3$)Nz4received an invalid combination of arguments - got (rJrK)r7r)rrr%sr:rUr)|sBu+r$s),,-RU 0D0D/EQHr<zsteps must be IntLike, got cg)Nz$number of steps must be non-negativerSrSr<r:rUr)s%Kr<)r%r9r&rrrWrrG)rr)!r@r(rcrnr r+r1rRrrrUrrrrr7rrrropsrr$rrrrrsrUr$rtrr)rrr%rr9r%r&rfactory_kwargs empty_tensorr dtype_redrrcast_rgrrr0s```` @r:rrOs%(( IIK1  P (/Fv/NO#~&& GGIN P &c+B6+JK||LU8KLLL % A A  # # %!  =)E LL&&u-z  2002 eU[[ ) )>tE{mLMM 5'" H eW % %:4;-HII LL!KL & N  zzzAA.AA z e^ , , ;;xOuOOL99>>&&..|UC C::uhMUMnM M a 1. 1B  " "5 ) )U-C-CE-J-J   !11 & + +Y-5FGG %KEAI &D ++ UQY wr{"" dWeai2-... C #3 ..r<r$c Uc[R"5n[RR U5(a[ U[ 5(a [U5nOI[ U[5(a4[R"UR5S:HS5 [X5n[ U[ 5(a [U5nOI[ U[5(a4[R"UR5S:HS5 [X5n[R"SXU455(a.[R"[R"55n U nSn O[Rn [ U[ 5(a[#S[%U535eUS:a[&e[R("UUUU UUUUS9n [[R*"X;5U5$)NrcgNz:logspace only supports 0-dimensional start and end tensorsrSrSr<r:rUlogspace..Tr<cgr;rSrSr<r:rUr<r=r<c3B# UHn[U[5v M g7frKr,rs r:rPlogspace..r.rzbase must not be complex, got r)rcrrrrr@rrr(rnr r+rRrrUrrrr7rrr) rrr%r$rr9r%r&rr0_dtyperets r:r r s }'') {{##E** eY ' 'ENE ~ . . LL q T ,E9E c9 % %#,C ^ , , LL QT *#5C||LU8KLLL % A A  # # %! &$  =d4j\JKK ax!! ..   # C #599T#7 ??r<indexingcgrKrS)rrCs r:rrs r<cgrKrS)rCrs r:rrsr<cv^^[US[[45(a4[U5S:wa[ S[U535e[US5n[ R "[R"SU55S5 [ R "[U5S:S5 [[U5S- 5Hon[ R "XRXS-R:HS5 [ R "XRXS-R:HS5 Mq S nTS :Xa&[U5S :nU(aUSUS/US SQ7nO[ R "TS :HU4S j5 /nUHm[T[5(d[ S[T535e[ R "TRS:H=(d TRS:HU4Sj5 URTR!55 M /n[#U5Hzunm[T[5(d[ S[T535eTRS:XaTR%S5mUR[&R("TXB455 M| U(aUSUSsUS'US'U$)NrrWz*Expected exactly 1 tensor list/tuple, got c3B# UHn[U[5v M g7frKr@r'rs r:rPmeshgrid..s@1Z:..rcg)Nz)meshgrid expects its inputs to be tensorsrSrSr<r:rUmeshgrid..rr<cg)Nz'meshgrid expects a non-empty TensorListrSrSr<r:rUrK s+Tr<cg)Nz3meshgrid expects all tensors to have the same dtyperSrSr<r:rUrKsIr<cg)Nz4meshgrid expects all tensors to have the same devicerSrSr<r:rUrKsJr<FxyrGijc>ST3$)NzDtorch.meshgrid: indexing must be one of "xy" or "ij", but received: rS)rCsr:rUrKs!!) ,r<zexpected TensorLike, got c>ST3$)NzEtorch.meshgrid: Expected 0D or 1D tensor in the tensor list but got: rS)rsr:rUrK)s[\][^_r<r@)r@rorhrZrrcrnrRrrmrr9r'r7rSrrrlr rrV)rCrrMswap_first_and_second_tensors result_shapegridsrs` @r:rrs^'!*tUm,, w<1  S[T5S[T5S3$)Nz5movedim: Invalid source or destination dims: source (z> dims) should contain the same number of dims as destination (z dims)ro)rWrVsr:rUmovedim..Ns) V ~''+K'8&9 Ar<)rrxc">S[T5S3$)Nz#movedim: repeated dim in `source` (rKrZ)rVsr:rUr[_s5d6l^1Er<c">S[T5S3$)Nz(movedim: repeated dim in `destination` (rKrZ)rWsr:rUr[cs:4 ;L:MQOr<rrW)r7rrcrnrZrSrhrr<setdictrrmrArr)rrVrWrssdssssdssrrXsidirCrs `` r:rr=sh F|s KC"n  LL F s;''  ::D u&&D&A BB u&&D+F GB b'C b'C LL B3s8E LL B3s8O S[A D BDk EE"I = KKN)a) KKO !GB]]5%+ .F Mr<)rr9r%rr&rc [R"U5 [R"U5 [R"U5nUc[R "5OUnUc[R "S5OUn[R"UUUUUS9$)Nrr) rrrrrcrr9rr)r[rrr9r%rr&s r:rr|sz v :&  , ,U 3E).E # # %EE$*NU\\% F    #  r<valc[R[R[R4H+nU[R"U5R ::dM)Us $ [R $rK)rcrmuint16riinforkrs)rgpossible_dtypes r:_strength_reduce_integerrlsI ;; ekkB %++n-11 1! !C ;;r<nrc ^^TcTm[R"TS:U4Sj5 [R"TS:U4Sj5 [Rn[T[R 5(a4[T[R 5(a[ [TT55n[R"TXtSS9n[R"TXtSS9n URS5U :Hn U[RS4;a4U(d-U RU=(d [R"55$[R"SUUUUSS 9n [R"XS5$) z' Reference implementation of torch.eye Nrc>ST3$)Nz%n must be greater or equal to 0, got rS)rmsr:rUeye.. #H!Lr<c>ST3$)Nz%m must be greater or equal to 0, got rS)rsr:rUrprqr<Frr]r@r)rcrnrsr@rIntWithoutSymIntrlrkrrstridedrrr&r) rmrrr%r9r&r range_dtyperange_nrange_mrones `` r:rrs y  LLaLM LLaLM++K!U++,,Au?U?U1V1V.s1ay9 ll1KeTGll1KeTG   R G +D %--&&zwwu9 7 7 9::jj !  {{4a((r<c  [R"U5 [R"U5 UbUO[R"[ U55nUbUO[ R "S5n[UUUUUUS9n[ R"Xq5$)Nrr) rrrrnr7rcr9rrH)r[rrr%r9r&res r:rrs~ v :&&EE,?,?Z@P,QE)Vu||E/BF  #  A ::a $$r<rr%r9r&rrc L[R"UUUUUUUS9n[X5$)Nr{)rcrrH) rsrrr%r9r&rrrzs r:rrs6  ## A  r<c [R"UU=(d UR[R:XaSOSUUUUUUS9$)NFrr{rcrrrrsrr%r9r&rrs r:r,r,sC ?? "177uzz1q##  r<c [R"UU=(d UR[R:XaSOSUUUUUUS9$)NTrWr{r~rs r:r'r'sC ?? !!''ejj0a##  r<c [R"U5 [R"U5n[R"U5n[R"U5n[ R "USSUUUS9$)Nrr)rrrr9r)rrrdtype_or_defaultrrrd)rr9r%rr&r[r's r:r(r(1se :&  - -e 4F  " "5 )E  $ $V ,F <<  #  r<c[R"U5 [R"U5 UbUO[R"[ U55nUbUO[ R "S5n[R"XUS9$)Nrr) rrrrnr7rcr9rr))rsrr%r9r&s r:r)r)Lsc v :&&EE,?,?Q,HE)Vu||E/BF   qf ==r<lowhighc [R"U5 [U[5(d[ S[ U535e[U[5(d[ S[ U535e[ U5n[ U5n[U[R5(d[ S[ U535e[R"U5n[R"XX$XSS9$)Nzlow must be Number, got zhigh must be Number, got r1)rrrr9rM) rr&r@r!rr7rrcrcanonicalize_devicer_uniform_helper)r[rrrMrr9s r:rr`s  c6 " "7S {CDD dF # #8d EFF C.C T?D eU[[ ) )>tE{mLMM  & &v .F   Tv r<rc*^^[R"UR5m[U[5(a [ U5nOUR m[R"TS:HU4Sj5 URR SSS[RR5S4;=(a URR S:Hn[R"U=(d URUR:HS5 [R"UR5nU[La0[R"[R"UT5U4S j5 [X R5n[R"XU5nUR!5$) Nrc>STS3$)Nz@only supports a 0-dimensional value tensor, but got tensor with z dimensionrS) value_ndimsr:rUmasked_fill..sVWaVbblmr<cudaxpumpsr5rcg)Nz,Expected `value` to be on same device as `a`rSrSr<r:rUrsBr<c>STS3$)Nzcould not convert to type z without overflowrS)rsr:rUrs0 =NOr<)rrrr@r!r7rSrcrnr9_C_get_privateuse1_backend_namerr r+rr)rsrr value_type is_cpu_scalarrrrs @@r:rr{s.%%agg.K%  %[ ZZ  !O m HHMMuehh&L&L&NPUV W + !!U*    5U\\QXX5 B ((5 W   ' ' K @ O  $E77 3E D#A <<>r<cV[R"XU5nURU5 U$rK)rcrcopy_)rsrrrts r:rrs& !5)AGGAJ Hr<c [SXX#S9 [[R"[R"XX#US95R 55$)z, Reference implementation of torch.allclose ztorch.allcloser)rrr)rrrcrrr)rsrtrrrs r:r-r-s@+qDL  %--4iPQVVX r<cR[R"XSS9 [R"X5 URUR:wag[ UR UR 5H up#X#:wdM g UR 5S:Xag[[[X555$)NFrrT) rrcheck_same_dtyperSrr[rrrr)rsrtrOrs r:r.r.s} A5A 1  vvAGGQWW% 6&  wwyA~ BqH r<rc~US:Xa'Ub'[U[5(d[U5S::dUcSn[U[5(aU/n[U[5(a@Uc[ [ UR 55n[RRXX#US9$[RRXX#US9$)NfrorGr) r@rrZrrhrmrSrcr matrix_normr)rr r r rs r:rcrcs U  z#s';';s3x1}  #se!S ;ejj)*C||''#e'LL||''#e'LLr<c^[R"TRS:HU4Sj5 [R"[R"TS55$)NrGc">STR3$)Nz+expected a matrix, but got tensor with dim rrsr:rUtrace..s=dii[Ir<r)rcrnrSrrrs`r:rwrws; LL QI 99UZZa( ))r<cX^S[[-S[[-S[4U4SjjnU$)Nrsrtr{c>T"X5$rKrS)rsrtbase_ops r:rop_make_r_binary_op..ropsq}r<)r(r")rrs` r:_make_r_binary_oprs3 J & J &  Jr<rcr[R"URS:S5 URSSup#[R"X0R S9R S5[R"X R S9R S5- U:n[R"X@5R5$)NrGcg)Nz2triu: input tensor must have at least 2 dimensionsrSrSr<r:rUtriu..Qr<rr8r] rcrnrSr[rr9rrrrrsrhrirs r:rr  LL ! Q 7723.!rr<rr8r]rrs r:rrrr<rrcUS:XdUS:XagUS:a[USU-5O[X-S:5n[S[XU-55n[S[XU-55nXC- S-nX4-U-S-nXV- n[SX-5n XyU4$)NrrrrrWrG)r rrk) rrr m_first_row m_last_row n_row_alln_row_trapezoidtrapezoid_sizediff_rowrectangle_sizes r:_get_tril_sizesr3s ax3!8*01*#c1v:&#clQ>N:OKQCv./JAs3f -.I .2O"./AQFN*HHN+N ; 66r<c^^^^[R"TS:U4Sj5 [R"TS:U4Sj5 [R"T[R[R4;UU4Sj5 g)Nrc>ST3$)Nzrow must be non-negative, got rS)rsr:rU_trilu_checks..M %CC5#Ir<c>ST3$)Nzcol must be non-negative, got rS)rsr:rUrNrr<c>STSTS3$)Nrzz" not implemented for ''rS)rrsr:rUrQs"TF25';r<)rcrnrrs)rrrrr%r&s```` r: _trilu_checksrEsO LLIJ LLIJ LL %++u{{++;r<rcl[SXX4U5 [XU5upxn [SU*5n [[R XEUS9n U "SU[R S9n U S- n [R"U *[R"X-SU --5-5n[R"U SU -S- U-U-S-- 5n[X-U5n[X5nU "SXS9nUU-X- S-U --nUU-n[R"[R"UU45[R"UU4545$)Nrrr%r9r&rrrGrW) rrrkrrcrrrJrsr+rr)rrrrr%r9r&rrr row_offset arange_kwxs1rt row_inds1 col_inds1xs2 row_inds2 col_inds2s r:rrVs7.#E:F2A#F2S/NKQJ VzI A~U]] ;CcA QBAEAGO!<<=I C1{?Q#6#Bi"ORU"UUVI' (>FI' 9I A~ 3Cs c/!3j@AIc I ;; Iy) *EIIy)6L,MN r<cUS:XdUS:XagUS:a[SX- 5OUn[S[X*5U-5n[XUS- 5upVnX-XV-- nX- n XU4$)NrrrW)rkr r) rrrrrtrapezoid_size_trilrectangle_size_trilr triu_sizers r:_get_triu_sizesrs ax3!8*01*#a&#KCW-34N3B#FUVJ2W/a 0FGI/N ; 66r<cv[SXX4U5 [XU5upxn [SU5n [[R XEUS9n U "SXS9n X-n X-nU "SU[R S9nSU - n[R"U*[R"UU-SU-- 5- 5n[R"USU -S- U- U-S-- 5n[UU5n[UU5nU(aUX--nUU -n[R"[R"U U45[R"UU4545$) NrrrrgrGrWr) rrrkrrcrrrJrsr+rr)rrrrr%r9r&rrr col_offsetrrrrrrtrrs r:rrs:.#E:F2A#F2S/NKQJ VzI A~ 3C I I A~U]] ;C {A QBAEAGO!<<=I CA Oa$7)$Cy#PTW"WWXI' 59I' 59I !67 J&I ;; Iy) *EIIy)6L,MN r<) out_int32r boundariesrrc ^[R"TR5S:HU4Sj5 [U[R5(aUO[R "U5nU(a[R O[RnTRSnUS:Xa[R"U5$[R"URUR[RS9nXe-nXgU- S--nTUn U(aX:n OX:n [R"XUS-5nUS:a[R"U[RS9n [[ R""U55n [%U 5Hsn [R"X-X5nXg:n [R"XXv- S--S5nTUn U(aX:n OX:n [R"U )U -US-U5nMu UR'US9$)NrWc,>STR5S3$)Nz2boundaries tensor must be 1 dimension but got dim(rKrp)rsr:rUbucketize..sDZ^^EUDVVWXr<r]rrrGr)rcrnr r@rhrrrsr[r,r+r9rr'rrr9r^rmr)rsrrr out_dtype n_boundariesrrmidmid_valcond_mid cond_updatenitersrs ` r:r/r/s LLAX 5<< ( (ell1oA( ekkI##B'Lq"" KK DE  C ;1$ $CoG ;< KKq 1Eaooauzz: TYY|,-vA++h4c?C+K++kCKA3E+EqIC oG";"<KK(k 937EJE 88)8 $$r<c^^Ub [S5e[R"[R"TR 5(+=(aQ [R "TR 5(+=(a% [R"TR 5(+U4Sj5 [R"TS:U4Sj5 UT[R"[R[R"T5S- -5--$)N"generator is not supported in refsc">STR3$)NzwCauchy distribution is a continuous probability distribution. dtype must be a floating point but you specified rrsr:rUcauchy..::>**Gr<rc>ST3$)Nz-cauchy_ expects sigma > 0.0, but found sigma=rS)sigmasr:rUr s ?wGr<r) rrcrnrrrrrrur9r: rand_like)rmedianr generators` ` r:rrsABB LL  " "4:: .. 3&&tzz2 2 3&&tzz2 2 G  LL  G EEIIdgg1F1L&MNN NNr<c^^Ub [S5e[R"[R"TR 5(+=(aQ [R "TR 5(+=(a% [R"TR 5(+U4Sj5 [R"TS:U4Sj5 [R"T5n[R"UR 5RS- nUSU- :n[R"XT*[R"U55nST- U-$)Nrc">STR3$)Nz|Exponential distribution is a continuous probability distribution. dtype must be a floating point but you specified rrsr:rUexponential..rr<rc>ST3$)Nz4exponential_ expects lambda > 0.0, but found lambda=rS)ratesr:rUr"s FtfMr<rGrr]) rrcrnrrrrrrrrQrr\)rrr uniform_valepsilonr  log_uniforms`` r:rFrFsABB LL  " "4:: .. 3&&tzz2 2 3&&tzz2 2 G  LL s M //$'K kk+++,0014GsW},I++i599[3IJK 9{ ""r<c^^Ub [S5e[R"[R"TR 5(+=(a% [R "TR 5(+U4Sj5 [R"ST:=(a TS:U4Sj5 [R"[R"[R"T5*5[R"T*5- 5S-$)Nrc">STR3$)Nzgeometric not implemented for rrsr:rUgeometric..?s0 =r<rrWc>ST3$)Nz0geometric_ expects p to be in (0, 1), but got p=rS)r sr:rUrCs B1#Fr<) rrcrnrrrrrJr]rr9)rr rs`` r:rLrL2sABB LL  " "4:: .. 3&&tzz2 2=  LL A!a%F ;;u{{EOOD$9#9:TZZ^K Lq PPr<c^^Ub [S5e[R"[R"TR 5(+=(aQ [R "TR 5(+=(a% [R"TR 5(+U4Sj5 [R"ST:U4Sj5 [R"T[R"T5-U-5$)Nrc">STR3$)Nzlog_normal not implemented for rrsr:rUlog_normal..Us1$**>r<rc>ST3$)Nz-log_normal_ expects std > 0.0, but found std=rSrsr:rUrYs ?uEr<) rrcrnrrrrrrD randn_likerrrrs` ` r:r`r`HsABB LL  " "4:: .. 3&&tzz2 2 3&&tzz2 2>   LL CE 99S5++D11D8 99r<)rr)rrr%r9r&c  ^Ub"U[R:wa[SU35e[T[5(d[R "TS:U4Sj5 Uc[ SUT455n[R "[U5S:S5 [R "USL=(a USLS5 [SU56nUSRnUSRnO|[R "[U[5(+=(a [T[5(+S5 Uc[R"5OUnUc[R"S 5OUn[R"US S UUS US 9n TU -U-$)Nz*layout must be None or torch.strided, got rc>ST3$)Nz)normal expects std >= 0.0, but found std rSrsr:rUnormal..ys I#Or<c3T# UHn[U[5(dMUv M g7frKrHrys r:rPnormal..}sL;a*Q 2K;s( (cg)NzFnormal expects that either mean or std is a tensor, or size is definedrSrSr<r:rUrs\r<cg)Nz.Cannot pass layout, or pin_memory without sizerSrSr<r:rUrsDr<c38# UHoRv M g7frKrKrys r:rPrs"normal expects mean and std to be scalars when size is definedrSrSr<r:rUrsTr<rrrF)rrrr9rr)rcrtrr@r'rnrhrZrurr9rrrd) rrrQrrr%r9r&rnormal_sampless ` r:rdrd`s^(f 5I&RSS c: & & 1HO  |LD#;LL L1  \   dN 1zT1 D !".rFr<gcܥL@rcrnrrr)rM_180_PIs r:rhrhs6 LL  " "4:: ..?CH ?r<c[R"[R"UR5(+S5 SnX-$)Ncg)Nz-deg2rad is not supported for complex tensors.rSrSr<r:rUdeg2rad..rFr<g9RFߑ?r )rM_PI_180s r:r?r?s6 LL  " "4:: ..?EH ?r<c*US:gRU5$r2)r)rr s r:r>r>s AI??3 r<cv^^[R"TR5S:H=(a TR5S:HUU4Sj5 [R"TRTR:HUU4Sj5 UU4Sjn[R"TR 5TR 5:HU5 g)NrWcN>STR5STR5S3$)Nz1D tensors expected, but got zD and z D tensorsrpr6rsr:rU_dot_check..s!/ |6%))+iXr<c<>STRSTR3$)Nz:dot : expected both vectors to have same dtype, but found rrrsr:rUrsL ::,eEKK=*r<c >STR5STR5STR5STR5S3 $)Nz+inconsistent tensor size, expected tensor [z ] and src [z.] to have thesame number of elements, but got rz elements respectively)rrsr: numel_error_dot_check..numel_errorsL9$**,{SXS^S^S`Rab004 ~U5;;=/Qg i r<)rcrnr rr)rr6rs`` r: _dot_checkrsx LL  a,EIIK1,X  LL ekk! *   LL. [X5 T"X5$rK)r)rr6rs r:wrapper#_dot_check_wrapper..wrappers4$r<)r )rrs` r:_dot_check_wrapperr!s  2Y Nr<rcUR5(aUR5(a{UR5(aA[R"UR 5UR 55R 5$[R "UR 5U5$UR5(a%[R "UR 5U5$X-R 5$rK)r0is_conjrcrrrrrs r:rrs  <<>>}}yyejjl;@@BBzz$))+u55 ]]__::ejjlD1 1 L   r<c,UR5(d[R"X5$UR5(amUR5(a3[R"UR 5UR 55$[R"UR 5U5$UR5(a2[R"XR 55R 5$UR 5U-R5$rK)r0rcrr#rrr;rrs r:rrs ??  yy%% ||~~ ==??::ejjlDIIK8 899TYY[%0 0 yyzz|,1133    5 ( - - //r<c[R"URU5nS/UR-nSXB'US:aX0RU-n[R "URUUR S9RU5U:Hn[R"X5RUR5n[R"XQU5$)NrWr]rr8) rrrSr[rcrr9r rrr)rOrr r; mask_shapers r:select_scatterr's   -CqvvJJO qy $ << QXX 6 ; ;J G5 PD //# # * *177 3C ;;t! $$r<)rdcV[U[R[R45$rK)r@rc TypedStorageUntypedStorage)objs r: _isStorager,s cE..0D0DE FFr<cSn[U5n/n[U[[45(a[ U5nU(aXQR -nUR U5 [ U5U:a"[S[U5RS35eUS:XaU$USnUn[U[[45(aMU$![a# [S[U5RS35ef=f)Nztoo many dimensions 'rrz.could not determine the shape of object type ') r,r@rorhrZitemsizerror7rr)seq scalar_typeMAX_DIMS is_storagerVr.handles r:_compute_sizesr5sHCJ E S4- ( (S  ++ +F V u: 4T#Y5G5G4HJK K Q;  L VF  S4- ( (" L  @cASAS@TTUV  s B==-C*c[U[5(a[R"5$[U[5(a%[U[ 5(d[R $[U[5(a[R $[U[5(a[R"5nU[RLa[R$U[RLa[R$U[RLa[R$[S5e[U[R 5(a UR"$[U[$5(a"['S[)U5R*S35e[U[,[.45(aSn[1U5nUS:Xa[R"5$[3U5HFnXn[5U5nUb[R6"X&5nOUnU[RLdMDUs $ U$[S[)U5R*35e)Nz'invalid default scalar type for complexznew(): invalid data type 'rrzCould not infer dtype of )r@rrcrrrrsrrrcfloatdoublecdoublehalfchalfrjrhrr TypeErrorr7rrorhrZrm_infer_scalar_type promote_types)r+ default_dtype scalarTyper.rMcur_itemitem_scalarTypes r:r=r=s#y!!&&((#w 3(=(={{#x  zz#w//1 EKK '<<  ell *== ejj (;; HI I#u||$$yy#s4T#Y5G5G4HJKK#e}%% S Q;**, ,vAvH 1:O%"00M , U]]*"!  2493E3E2FG HHr<r@r+c [U[5(aBUR5S:Xa.UR5R USSS9R S5$[U[5(aUR5R USSS9$[U[ 5(a[R"XS9$UnU(d[R"S5$[R"UVs/sHn[X5PM sn5$s snf)NrWrT)rr9r$rSrr) r@rhrdetachrr r!rcr)rr_recursive_build)r@r+r0rs r:rErEs#v399;!#3zz|ZDINNrRR C zz|ZDII C ""399 C  A[[M*:<M NNs(DcB[U[R5(an[R"U(+S5 UnU(aUR 5nU(a UR OUn UbUO UR n URU U SUS9$[US5(a[$UbUOUSn U(a [U5OUn [U5(a[$[R "U 5RS:Xa[$[X5n U RXSSS9n U $)Ncg)Nz,Can't pin tensor constructed from a variablerSrSr<r:rU)_internal_new_from_data..s$Rr<F)r9rr'r$__cuda_array_interface__r9meta)r'r$)r@rcrhrnrDrr9rhasattrNotImplementedr=r,r7rE) optionsr1 device_optdatacopy_variables copy_numpytype_inferencer&rinferred_scalar_typer9rs r:_internal_new_from_datarTs$ %% NR  **,C,:syy )53::vv&   t/00&1Zwx7HF7E-d3;$ <<  $ $ .! ! ""6=6eRWX Mr<)rr9r&rc $[U[R5(a[R"S[ SS9 USLn[ SS0UbUO[R"5UUSSUUS9nUR5 U(aURU5 U$)NzTo copy construct from a tensor, it is recommended to use sourceTensor.detach().clone() or sourceTensor.detach().clone().requires_grad_(True), rather than torch.tensor(sourceTensor)rGrr9rT)rPrQrRr&) r@rcrhrr UserWarningrTrdetach_r)rOrr9r&rrR new_tensors r:rr-s$ %%  l   d]N( 5"(?(?(A % J!!-0 r<rK)F)rNN)gh㈵>g:0yE>Fr@)NN)NNNNNFF)FFN)NF)NNFr)rr]) NNNTreflectFNNN)NNNTFNNF)rS)rWr')rrrW)rrr])rNrW)r+)rr)rNF)rrWNr~)rWrGN(rRrrrr9rrcollections.abcrrrenumr functoolsrrr r typingr r r rrrc torch._prims_primsrtorch._prims_common _prims_commonrtorch.utils._pytree_pytreerbrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*torch._prims_common.wrappersr+r,r-r.r/r0$torch.testing._internal.common_dtyper1rrhr DispatchKey_opsr4rr;rDrur torch._decomprobjectrrrrrr2 INT_TO_FLOATr3r4r6r5r7r8DEFAULTr9r:r0r;r<r=r@rArBrCrDrErG NO_OPMATHrHrIr*rJrKr ALWAYS_BOOLrSrTrUrVrWspecial multigammalnrbrXrYr[r\r]r^r_rrrarrrer$rfrgrrirjrrkrlrmrnrorprqrs BOOL_TO_LONGrtrurvrxrrlroryrzr{r|r}r~rrrrrrrrrrrrrrrhrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr ScalarSelf ScalarOtherScalarself_outrr)rrrr%r+r-registerr9r_r1r3r5r8r?rrZrkrrr dim_IntList IntList_outrorrrrrrrrrrrrrrrrrrrpy_implCompositeImplicitAutogradMetarrrrrr`rrrrrrrrrrrWr[rr _subclasses fake_implsregister_op_implrrr1r2rr3rrrrrrrrr.rrrrrrrrNrOrQrRrRrMrPrrXrrrr rrrrrrrrrrrrrrrrrrrr r rrrtrrr!r"r+r%r&r$r#rrr start_step start_outrZrr rrrrlrrrr,r'r(r)rrrr-r.rcrwrrrrrrrrrrqrrr/rrFrLr`rdr rhr?r>rr!rrr'abs_acos_acosh_add_addcmul_addcdiv_asin_asinh_atan_atanh_atan2_ bitwise_and_bitwise_left_shift_ bitwise_not_ bitwise_or_bitwise_right_shift_ bitwise_xor_ceil_clamp_ clamp_min_ clamp_max_conj_physical_ copysign_cos_cosh_cumsum_cumprod_deg2rad_digamma_div_eq_erf_erfc_erfinv_exp_exp2_expm1_ float_power_floor_ floor_divide_fmod_frac_gcd_ge_gt_ heaviside_hypot_igamma_igammac_i0_lcm_le_lerp_lgamma_log10_log1p_log2_log_ logical_and_ logical_not_ logical_or_ logical_xor_lt_mul_ mvlgamma_ nan_to_num_ne_neg_ nextafter_pow_rad2deg_ reciprocal_ remainder_rsqrt_sgn_sigmoid_sign_sin_sinc_sinh_sqrt_square_sub_tan_tanh_tril_triu_ true_divide_trunc_xlogy_cauchy_ exponential_ geometric_ log_normal_zero_rrrrrrrrrrr r r,r5r=rErTrtorch._refs._conversionstorch._refs.ffttorch._refs.linalgtorch._refs.nn.functionaltorch._refs.specialrSr<r:rsP 88<<77 #$$662I l \ hh""  zz~~ 2 Rj9=8Cx1     D  2##44 ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#J#JK L ##++ (/n/)/**+ &&,& ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ  #7AA  N  :  .     ^ J >  " ##>#####++ ##++!N!~! ! N~##// **>* *##B#N#NO1^11P1##//11>1 1##//11>1 1##B#N#NO^P u}}11: >##// 0n00 0##00$''##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ##B#O#OPQ  !%XX X ;; X XX' #!7DD :?, ,',26,, ( ,"( ! $ $  d        )6##++    >  N ~ ##B#O#OPQ # #7??12J^J#JnJ $ J##B#O#OPQ##B#O#OP/~/./Q/##++ ##++ ##// ##B#O#OPQ ##B#O#OP4Q4 ##B#O#OPQ##B#O#OPQ##00 n ##B#O#OPQ##B#O#OPQ##++  !.!^!L  ## %"&66r( "('5 'B( [[(! #"7?? $ 8 "8 "8   8 " 8:$7DD$$   $7??#>#n###$7??"."^"""$7??"."^"""$7??.>.n...$7??#>#n###$7DD$2+20K2 2!  !%  " ": "($7CC$.^ $7DD " "6  " " t$==EE $"' ONZ/ONZ4OO  O>FVFF6F161f112$7??$$ N~.  $7??$$ N~.  $7??$# N~.   # Z$775)G#H7%$7$7??$$ >n  $7CC$.^ $7CC$.^ $7??$$ ^^  $7??$$ ^>  $7DD$$ nN  $7DD$$ ~.^         2 ((( (  (  (  (V$7??$$ Q> Qn Q  Q $7CC$.^ $7??$$ TTNT~T  T0$7??$$ ,. ,^ , ,  ,$7CC>n##B#N#NO>P $7CC.^$7CC>n$7CC$.^ $7??~.^$7??~.^$7?? $>n $7CC$.^ $7AA$$ !!N!~!  !$7??!!N!~!!  "  ( " ( " (  (# (! #"7?? $8 "$8 "$8  $8 " $8N$7DD   $  >n   # #"7DD:^j (:^j-H: $ :.$==EE  $ "NZ/"NZ4O"  " % #67DD , , ,,  ,  , & ,* % #67?? , , ,,  ,  , & ,* # #+7?? *.)-  $ &  $ &   $  6' *.& & $ &&&(&' *.& & $ &&&(& **+..E&#+$6.,. ( --. ../ ))* ++, #"7AA (,'+# #$#$# -+0/)#. # ?D?T?T).)<)<$9=v& "&N&z&#&%) $"&04# T ! ;;  +  LL4   &&- t  204  LL ;;   &&-   #s(^ "04   ;;   &&-   #s(^ "04  ;;    &&-   #s(^  04   &&-   #s(^ ."&"T"0 . n X"  $CC C C  C T/ CC ;; C $C2CCL<!     D   " !   D "$--txx/C/CDE#' !%  td3i  ;;   $ F:C C C, "#'   td3i  $ #: "    D $   #. "    D $   #.!     %)  DTk  T! "6! #'  1 %) 11 td3i 1Tk1 1 T! 11"1, " +  ++ D+++#+\ & VV !$(  DTk    T! '2 & VV    %)   D Tk   T! '  " #07?? 7A 7A 7A 7A  7A  7A7A # 7Av + (>2 2 +;I +eNC/00 +   # *8   + (>2 2 +;I +eNC/00 + + (>2 2 +;I +eNC/00 +&"& A A  A  A$J A  A//0  "& R R R R  R $J R  R1R22'' (M(MN'' (8(89O4#7O:OO1N1)11 ! #$7AAyG#yG#yGnyG " yGz #,###>,,- ?@P P $S P2<PP.Ph8=7N7N 7  7!&!4!4 7  7 #&#>##  $/4FfFFF%FTFv^Ss5QTAT;U,1~1#1C11* " N"2~#nn(+n(<FID "). 666 !"@ > tCy ^ ..667D D TMD 4-D D  D  DD D 666 !"D8DN../ VVV$L LL TML 4- L  L  666 !" L%0L^..t/E/E/M/MN$O$  %,~,,&, $ #. #.,#.36#.AO#.#.%#.N+??@"!  "&#'p? p? p?d p?d p? TM p?  p?p?p?Tkp?4Kp?D[p? p?Ap?hKAAB"!  Z Z Z d Z d Z TM Z  Z Z TkZ  $JZ  Z CZ | ",9<DGOR8 $ 51f5151%51pRR*.RRj8?N8?8?8?~<~N8>H8>H8>n8>#8>v # >>)9>>$>< !3   #  G% GC G G$ G" !%  ;;  * #&#>## #&#>##SScS)SS  $ n 3 /A % " *3z:  :  C   J  ( <<<$.<7AJ7N<)< (();;;$.;7AJ7N;*; // / /  " /  /d        "))* KJKSKK+K" ))* ~ HtO ~ + B--.56 #Cy/2 &\/DDFDF(*DF DF >3  DFP(-(-,4(- >3 (-X&+&+,4&+ >3 &+R &  1  1  1 1' 1--.           /0 & # # # #  #  #'#L(  5353 53 53  53  53)53p( !$$~"6!$>!$)!$H)n-).) (n ( (5G ('99(9$  N  #^$' * *s *# *. *( *   $+ +%(+03+;>++%+(() nCs*!%99 9 ;;  9 $990 $ !% TT T ;;  T $ T  T%T % !% UU U ;;  U $ U  U&U'33c3n3(3 ))*=N=Ii0@$@=^=+=*#.##N# ^  >B%D ||%D#ll%D14t%D \\%D%DP !% ==$().)@)@% ;; % LL% T ! %  %  %&&%%%P !% ==$( ;;  LL  T !   $'  !%"&$(  ;;   LL4   T ! (,../  !%"&$(   ;;   LL4   T !08 **+ !% ==$(  ;;  LL T !     ,0'  !%"&$(  ;;   LL4   T ! (2 ))* !% ==$(  ;;  LL T !     +0 &  !%"&$(  ;;   LL4   T ! '2 &  !%"&$(  ;;   LL4   T !'0 ==>15  &&- ?( !%$("&).)>)>&& ;; & T ! & LL4  &  &&&&&&)&R//1F1FGH !c !% ==$(c c d c c ;;  c LL c T !ccccIcL " #27??88V8Vj-@8 # 84 &  !%$( ==X/  &X/ n $X/ X/ ;;  X/ T ! X/ LLX/X/X/X/'X/v &   9@ !%$( ==9@  &9@ n $9@ 9@  9@ ;;  9@ T !9@ LL9@9@9@9@'9@x  h~. #     ~     &CtN33eN6KKCC .C'CN; ; " ";''; ;~**+  !%$( == uY' '  ;;   T !  LL ,6#%++! %)!% ==$(%) %) Tz%) ;;  %) LL %) T ! %)%)%)%)"%)V**DIIMM:;  !% ==$(% %% ;;  % LL % T ! %%%%<%>!%"&$().)>)> ;;   LL4   T ! &&.( !%"&$().)>)> ;;  LL4   T !   &&),' !%"&$().)>)> ;;  LL4   T !   &&(, **+ !%$("&  ;;  T ! LL4      ,8!% ==$( > > ;;  > LL > T ! >  > >,""  e  *u    ;;  6(() *>**@V***Z))*  + 4J  +         "^4& " " M !% M M s{TM DM M ;;  MM#M2 # **>*$* [ ) l +  "  3N 3c 3. 3# 3 "  3N 3c 3. 3# 3(77377c3m8L7$        ;;  LL   "))* "  =="" " " " ;; " LL " """+"R77377c3m8L7"))* &  =="& & & & ;; & LL & &&&+&R'   7%7%7% 7%  7%(7%t $ #!7??O % O"(() #!7??# * #8' #!7?? Q ( Q ( #!7?? : ) :$ $ #8??   2'    2'%2'j %HdH&H##B#O#OP.Q##B#O#OP.Q**+  X_ , =*! #)7??  "   " #)7?? 0 # 0 ++, %n%>%%C%-%Sd u S  !  !d u d u  u [) #$67[) J' $%89[) d u  9 % 9 % }- ( # Sd    !  !  !SBSd  Sd u [) u l+ ddSBB 9 % u     !BSBd   u  u dS[) [) J' [) BS ( # J' BS 9 % S  !J' 9 % u S  !dSddd  SSddd[) u  u   [) 9 % J' d !$** - &t7$T]]3 "4;;/ #4;;/ #DLL1 #DLL1 dff %%dnn5"4;;4P %dnn5  + G 4/Ih "8;M"MH5rtUL  r<