Nj|~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 /S QrD\RrE\RRrG\RRRrJS rKS rLS rMS S.SjrNSSKOJPrP \Q"5rR\RSSS.S\4SjjrSSrTSrU\S"\,RS S9S5rW\S"\,R5S5rY\S"\,R5S5rZ\S"\,R5S5r[\S"\,R5S5r\\S"\,R5S5r]\S"\,R5S5r^\S"\,R5S5r`\S"\,RS S9S 5ra\P"\JR5S!\:4S"j5rb\P"\JR5\C"5S!\:4S#j55rc\S"\,R5S$5rd\S"\,R5S%5re\S"\,R5S&5rf\S"\,R5S'5rg\S"\,R5S(5rh\S"\,R5S)5ri\S"\,R5S*5rj\S"\,R5S+5rk\S"\,R5S,5rl\C"5\A"S-\,RS.9S/\:S0\4S\:4S1j55rnS/\:S0\4S\:4S2jro\P"\JR5\C"5S!\:S\:4S3j55rp\S"\,RS S9S45rq\S"\,RS S9S5\:S\:4S6j5rrS/\:S\:4S7jrs\S"\,RSS89S/\:S\:4S9j5ru\S"\,R5S/\:S\:4S:j5rv\S"\,RS S9S/\:S\:4S;j5rw\S"\,RS S9S/\:S\:4S<j5rx\S"\,R5S/\:S\:4S=j5ry\T"\RRS>5r|\S"\,RSS89S/\:S\:4S?j5r}\S"\,R\JRS89S@5r~\S"\,R5SA5r\S"\,R5SB5r\S"\,R5SC5r\S"\,R5SD5r\S"\,R5SE5r\C"5GSS/\:SF\SG\\GR S\:4SHjj5r\P"\JGR5\C"5\A"SI\,RS.9GSSJ\:SF\*SK\S\:4SLjj555r\P"\JGR5\C"5GSS/\:SM\\4SN\\4SO\\4S\:4 SPjj55rS/\:4SQjr\S"\,R\SR9SS5rS/\:S\:4STjrS/\:S\:4SUjr\S"\,R5SV5r\P"\JGR5\C"5\A"S-\,RS.9SSW.S/\:SX\S\:4SYjj555r\S"\,R5SZ5r\S"\,R5S/\:S\:4S[j5r\S"\,RS S9S\5r\S"\,RS S9S]5r\S"\,RS S9S^5r\S"\,R5S_5r\S"\,R5S`5r\S"\,R5Sa5r\S"\,R5Sb5r\S"\,GR2SS89S/\:S\:4Scj5r\S"\,R5Sd5r\S"\,R5Se5r\S"\,RS S9Sf5rSJ\:S\:4Sgjr\RSS S S SS 4S\4Shjjr\P"\JGR@5\C"5\A"Si\,RS.9SSj.S/\\:\44Sk\\:\44Sl\\44Smjj555r\"\,RSSSn9So5r\"\,RSp9S/\:Sk\:S\:4Sqj5r\"\,RSp9S/\:Sk\:S\:4Srj5r\"\,RSp9S/\:Sk\:S\:4Ssj5r\"\,RSp9S/\:Sk\:S\:4Stj5r\"\,RSp9S/\:Sk\:S\:4Suj5r\"\,RSSv9S/\\:\44Sk\\:\444Swj5r\P"\JGRP5\C"5SSx.S/\\:\44Sk\\:\44Sy\\4Szjj55r\"\,RSSv9S/\:Sk\:S\:4S{j5r\"\,GR2Sp9S/\\:\44Sk\\:\44S\:4S|j5r\C"5S/\\:\44Sk\\:\44S\E4S}j5r\"\RXRS SS~9S/\\:\44Sk\\:\444Sj5rS/\ESk\ES\E4SjrS/\ESk\ES\E4Sjr\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSS Sn9S/\:Sk\:S\:4Sj5r\P"\JGRf5\C"SS5SJ\:S\\:\:44Sj55r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSSv9S/\:Sk\:S\:4Sj5r\"\,RSSv9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S!\:S\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5rS\S/\:Sk\:S\S\SS4 SjrGSS/\:Sk\:S\S\S\S\:4 Sjjr\"\,RSSSn9S/\:Sk\:4Sj5r\"\,RSSv9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSp9S/\:Sk\:4Sj5r\S"\,R5S/\:4Sj5r\"\,RSp9S/\:Sk\:4Sj5r\"\,RSp9S/\:Sk\:4Sj5r\"\,RSSv9S/\:Sk\:S\:4Sj5r\"\,RSp9S/\:Sk\:S\:4Sj5r\"\,RSp9S/\:Sk\:S\:4Sj5r\"\,RS S9S/\:Sk\:S\:4Sj5r\"\,RSSv9S/\:Sk\:S\:4Sj5r\"\,RSSSn9S/\:Sk\:S\:4Sj5r\"\,RSp9S/\:Sk\:S\:4Sj5r\P"\JGR5\C"5GSS/\\:\44Sk\\:\44Sl\44Sjj55r\P"\JGR5\C"5\A"Si\,RS.9SSj.S/\\:\44Sk\\:\44Sl\44Sjj555r\"\,RSSS S9S/\:Sk\:S\:4Sj5r\P"\JGR5\C"5\A"Si\,RS.9S/\\:\44Sk\\:\444Sj555r\"\RXRSS S9S/\\:\44Sk\\:\444Sj5r\P"\JGR5\C"5\A"S\,RS.9SS.SJ\:S\:S\:S0\4S\:4 Sjj555r\P"\JGR5\C"5\A"S\,RS.9SS.SJ\:S\:S\:S0\4S\:4 Sjj555r\P"\JGR5\C"5\A"S\,RS.9GSS/\:S\\;S\\;S\:4Sjj555r\P"\JGR5\C"5GSSJ\:S\\;S\:4Sjj55r\P"\JGR5\C"5GSSJ\:S\\;S\:4Sjj55r\P"\JGRGR5\P"\JGRGR5\P"\JGRGR5\P"\JGRGR5\P"\JGRGR5\C"S S9\A"Si\,RS.9GSS\ES/\\;Sk\\;4Sjj5555555r\P"\JGR5\C"5\GRS.S/\:S\GRS\:4Sjj55rS S.S/\ESk\E4Sjjr\P"\JGR5S/\:S\44Sj5rGSS/\:S\\'SG\\GR S\\S\\GRS\\GRS\\S\S\4Sjjr\S5r\GRGSS\GRSG\GR S\S\S\\GRS\\\44 Sjj5r\GRGSS\SG\GR S\S\S\\GRS\\\44 Sjj5r\GRGSSG\GR S\S\S\\GRS\\\44 Sjj5r\GRGSS\ES\S\S\\GRS\\\44 Sjj5rS/\ES\4SjrS/\:S\:4SjrS S SSSSS.S/\:S\S\S\S\\*S\SG\\GR S\\ES\6S\:4SjjrGSSjr\P"\JGR5\C"5GSS/\:SF\\*SK\S\:4Sjj55r\P"\JGR5\C"5GSS/\:SF\\*SK\S\:4Sjj55r\P"\JGRGR\JGRGR/5GSSSS.S/\:SF\\\\\\4SK\SG\\GR S\\ES\:4 Sjjj5rS/\ES\E4Sjr\P"\JGR5GSSSS.S/\:SF\\\\\\4SK\S\\ES\:4 Sjjj5r\P"\JGR5GSSS.S/\:SF\\*SK\S\\ES\:4 Sjjj5r\P"\JGR5GSSS.S/\:SF\\*SK\S\\ES\:4 Sjjj5rGSSjr\P"\JGR5\C"5GSSS.S/\:SF\\*S\\SK\S\\4S\:4 Sjjj55r\P"\JGR5\C"5GSSS.S/\:SF\\\\\\4S\\SK\S\\4S\:4 Sjjj55r\P"\JGR5GSSSS.S/\:SF\\*SK\S\:4Sjjj5r\P"\JGR5\C"SS5GSSSSS.S/\:SF\\*S\\SK\S\\44 Sjjj55r\P"\JGR5\C"SS5GSSS.S/\:SF\\*S\\SK\S\\44 Sjjj55Gr\P"\JGR5\C"5\A"S\,RS.9SSS.SJ\:S\:S\:S\4Sl\4S\:4 Sjj555GrS\\:\ \:4S\:S\\:\\:S444SjGrS\SF\S\:S\:4SjGrS\\:\ \:4S\:S\\:\\:S444SjGrS\\:\ \:4S\:S\\:\\:S444SjGrGSS/\:S\7GS\8GS\\S\:4 GSjjGr\P"\JGR5\C"5GSS!\:GS\:S\7GS\8GS\\S\:4 GSjj55GrS\74GSjGr\JGRGRGR\GGR5\JGRGRGR\GGR5S\\:4GSj55Gr S/\:S\7S\:4GSjGr\P"\JGR5\C"5\A"GS\,RS.9GSGS \jj5Gr(\JGRTGRGR\GGR5GSS!\EGS4\GS5\\GS6\\GS7\\EGS8\GS:\GS;\\GS\\S\E4GS?jj5Gr*GS@\7GSA\8GSB\S\GSC\4 GSDjGr+\P"\JGRX5\C"5S/\ES\E4GSEj55Gr,S/\:GSF\S\:4GSGjGr-S/\:GSF\S\:4GSHjGr.S/\:GSI\7S\:4GSJjGr/SJ\:S\:S\:4GSKjGr0\P"\JGRb5\C"5GSS/\:GSL\*S\*S\:4GSMjj55Gr1\P"\JGRd5\C"5GSS/\:GSO\S\)S\:4GSPjj55Gr2GS \\C"5SSj.S5\9SF\GSZ\9GS[\9Sl\44 GSbjj5Gr?\P"\JGR5\C"5S5\9SF\GSZ\94GScj55Gr@\P"\JGRGR5GSS/\:SF\\*S\:4GSdjj5GrA\P"\JGR5GSSJ\EGSe\\SF\S\\E4GSfjj5GrCGSS/\:GSg\\E\*4SF\S\\:S44GShjjGrDS/\:GSg\*S\\:S44GSijGrES/\:GSg\*S\\:S44GSjjGrF\P"\JGRGR5\C"5GSSJ\:GSk\S\:4GSljj55GrG\P"\JGR5\C"5GSS!\:GS\:GSk\GSm\GSn\S\:4 GSojj55GrI\P"\JGR5GSSJ\:GSk\GSm\GSn\S\:4 GSpjj5GrJ\P"\JGR5\C"5GSGSX\:GSk\GSm\GSn\S\:4 GSqjj55GrK\P"\JGR5\C"5GS \\:S\:4GSrj55GrMGS \\:S\:4GSsjGrLS/\:GSt\*S\<4GSujGrN\P"\JGRGR5S/\:4GSvj5GrOS/\:S\:4GSwjGrP\P"\JGR5S/\:S\:4GSxj5GrQ\P"\JGR5S/\:GSy\GSm\S\:4GSzj5GrRG\RGrS\P"\JGR5SJ\:GSB\S\GSC\S\:4 GS{j5GrT\P"\JGR5\C"5SJ\:GSB\S\GSC\4GS|j55GrUSSS.S/\:SF\SG\\GR S\\ES\:4 GS}jjGrV\P"\JGR5SSS.S/\:SF\SG\\GR S\\ES\:4 GS~jj5GrW\P"\JGR5SSS.S/\:SF\SG\\GR S\\ES\:4 GSjj5GrX\P"\JGR5S/\:SF\S\:4GSj5GrY\P"\JGRGR5S/\:GSI\7\\7-S\:4GSj5GrZSJ\:S\:S\:4GSjGr[S/\:S\:4GSjGr\\C"5GSS/\RGS\RSF\\S\R4GSjj5Gr]\C"5S\GRSSS\GR&GS.SG\\GR S\GRS\\'GS\S\S\GRS\:4GSjj5Gr_\C"5S\GRSSS4SG\\GR S\GRS\\'GS\S\S\:4 GSjj5Gr`\P"\JGR5\C"5SSSSGS.S/\:S\7SG\\GR S\\GRS\\'S\S\:4GSjj55Gra\P"\JGR5\C"5SSSSGS.S/\:S\7GS\8SG\\GR S\\GRS\\'S\S\:4GSjj55Grb\P"\JGRGR5\C"5S\GRSSSGS.SG\\GR S\GRS\\'S\GS\S\:4 GSjj55Grc\P"\JGR5\C"5SSSSSGS.S/\:S\7SG\\GR S\\GRS\\'S\GS\S\:4GSjj55Grd\P"\JGRGR5\C"5S\GRSSSGS.SG\\GR S\GRS\\'S\GS\S\:4 GSjj55Gre\P"\JGR5\C"5SSSSSGS.S/\:S\7SG\\GR S\\GRS\\'S\GS\S\:4GSjj55Grf\P"\JGR5\C"5SSSSGS.S/\:S\7GS\4SG\\GR S\\GRS\\'S\S\:4GSjj55Grg\JGRGRGR\GGR5GSS\:S\:S\\GRS\:4GSjj5Grh\P"\JGR5\C"5SSSSS\GRGS.S/\:SG\\GR S\\'S\\GRS\GS\S\GRS\:4GSjj55Gri\P"\JGRGR\JGRGR/5\C"5GSS\GRSSSGS.GS\4GS\\4GSC\4SG\\GR S\GRS\\'S\GS\S\:4GSjjj55Grj\P"\JGR5\C"5\A"GS\,RS.9GS\EGS\EGS%\\E\444GSj555Grm\P"\JGR5\C"5SS\GRSSGS.GS\\4\:4GS\\4\:4GS\4SG\\GR S\\'S\GRS\GS\S\:4GSjj55Grn\P"\JGR5\C"5GSSS\GRSSGS.GS\\4\:4GS\\4\:4GS\4GS\4SG\\GR S\\'S\GRS\GS\S\:4GSjjj55Gro\GS \ \:GS\4GSj5Grp\GS \:GS\4GSj5Grp\P"\JGR5GS \\:\\:\\:4GS\S\\:4GSj5GrpS!\:GS\\\)4GS\\\)4S\:4GSjGrq\P"\JGR5\C"5SS\GRSSGS.GSI\\7\\74GS\8SG\\GR S\\'S\GRGS\S\S\:4GSjj55GrrGS\S\GR 4GSjGrs\P"\JGR5\C"5GSS\GRSSSGS.GS\GS\\SG\\GR S\GRS\\'S\GS\S\:4GSjjj55Grt\P"\JGRGR\JGRGR/5\C"5S\GRSSSGS.GSI\7GS\4SG\\GR S\GRS\\'S\GS\S\:4GSjj55GruSSSSS\GRGS.S/\:GS\4SG\\GR S\\GRS\\'S\GS\S\GRS\:4GSjjGrv\P"\JGR5\C"5SSSSS\GRGS.S/\:SG\\GR S\\GRS\\'S\GS\S\GRS\:4GSjj55Grw\P"\JGR5\C"5SSSSS\GRGS.S/\:SG\\GR S\\GRS\\'S\GS\S\GRS\:4GSjj55Grx\P"\JGRGR5\C"5SSSSSGS.SG\\GR S\\'S\\GRGS\S\S\:4 GSjj55GryS\GRSSGS.S/\4SG\\GR S\GRS\\'S\S\:4 GSjjGrzGSGSI\7GS\\\\4GS\\\\4GS\7SG\GR S\'S\:4GSjjGr{\P"\JGR5\C"5S/\:GS\:S0\;4GSj55Gr|\P"\JGR5S/\:GS\:S0\;S\:4GSj5Gr}GSS/\:Sk\:S\S\S\S\4 GSjjGr~S/\:Sk\:S\4GSjGr\P"\JGR5\C"S S9GSSGS.S!\:GS1\\\\4SF\\*SK\SG\\GR S\:4 GSjjj55Gr\P"\JGR5\C"5SJ\:S\:4GSj55GrGSGrG\"\5GrG\"\5GrG\"\5Gr\P"\JGR 5\C"5GSS/\:GS\S\:4GSjj55Gr\P"\JGR5\C"5GSS/\:GS\S\:4GSjj55GrGS\GS\GSk\S\\\\44GSjGrS\GS\GS\SG\GR S\GRS\4 GSjGr\P"\JGR5\C"5GS\GR\GRGSSGS.GS\GS\GSk\SG\GR S\GRS\'S\S\:4GSjjj55GrGS\GS\GSk\S\\\\44GSjGr\P"\JGR5\C"5GS\GR\GRGSSGS.GS\GS\GSk\SG\GR S\GRS\'S\S\:4GSjjj55Gr\P"\JGR5\C"S S9SSGS.S/\;GS\:GS\GS\4GSjj55Gr\P"\JGR5\C"5\A"SI\,RS.9GSGSj555Gr\P"\JGR 5\C"5\A"SI\,RS.9GSGSj555Gr\P"\JGR"5\C"5\A"SI\,RS.9GSGSj555Gr\P"\JGR$5\C"5\A"SI\,RS.9GSGSj555Gr\P"\JGR&5\C"5\A"GS\,RS.9GSSSSSSGS.GSjj555Gr\P"\JGR(5GSNSGS.GSjj5Gr\S"\,R5SJ\:4GSj5Gr\S"\,R5SJ\:4GSj5Gr\P"\JGR.5\C"5GSSF\\*4GSjj55GrGSGrGSGr\P"\JGR45\C"S S9G\\A"GS\,RS.9GS5555Gr\P"\JGR65\C"S S9G\\A"GS\,RS.9GS5555Gr\P"\JGR85\C"5S5\:GS\:SF\GSZ\4GSj55Gr\U"\W5Gr\U"\Y5Gr\U"\Z5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\[5Gr\U"\\5Gr\U"\]5Gr\U"\^5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\`5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\a5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\c5Gr\U"\5Gr\U"\d5Gr\U"\e5Gr\U"G\W5Gr\U"G\X5Gr\U"G\5Gr\U"\f5Gr\U"\5Gr\U"\5Gr\U"\g5Gr\U"\i5Gr\U"\h5Gr\U"\j5Gr\U"\l5Gr\U"\k5Gr\U"\5Gr\U"\q5Gr\U"\5Gr\U"\5Gr\U"\r5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\~5Gr\U"\5Gr\U"\5Gr\U"G\m5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\|5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"G\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"G\5Gr\U"G\5Gr\U"\5Gr\U"\5Gr\U"\5Gr\U"G\5Gr\U"G\5Gr\U"G\5Gr\U"G\5Gr\U"\p5Gr\"\JGR5Gr\"\JGR 5Gr\"\JGR5Gr\"\JGR,5Gr\"\JGR:5Gr\"\JGR5Gr\"\JGRL5Gr\"\JGR5Gr\"\JGR5Gr\"\JGRrS GS9Gr\"\JGR5Gr\"\JGR5GrGSGrGSGr GSGr GS\GR GS\\;\<44GSjGr GSGSjGr SSSSGS.GSjGr SSGKrSSGKrSSGKrSSGKrSSGKrg(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(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 e/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/torch/_refs/__init__.pyis_noncontiguous_supportedr:ps T> 1V[[E11cSnSSKJn UH!n[XC5(dMURn O [ U5(dUR 5nU$)Nr) FakeTensor)torch._subclasses.fake_tensorr= isinstance fake_devicer:r) input_tlistoutputr8r=rs r9handle_noncontiguous_outputsrCtsK F8  a $ $]]F  &f - -""$ Mr;c ^ ^ ^ ^ SSKJnJnJnJn [ R RRRn[S[SU555n[U5S:XagUH$m [T [5(aM[ST 5e 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_falsehas_hint is_nested_int size_hintc3V# UHn[U[5(aU4OUv M! g7fNr?r.0xs r9 $_broadcast_shapes..s*9A1g&&A-9s')c USL$rJrNs r9#_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 g7frJlen)rMshapes r9rOrPs4VESZZVz9Attempting 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 rR)arg_idx common_shapeidxrZsr9rTrUs:LUSVZLX\]`\ab55Ar;)%torch.fx.experimental.symbolic_shapesrErFrGrHtorchfx experimental_configbacked_size_oblivioustuplefilterrYr?r RuntimeErrorrmax enumeraterange_check ValueError) _shapesrErFrGrH backed_soshapesabr^r_r`rZs @@@@r9_broadcast_shapesrtsL%%--CCI /9F  6{a%**_  4s4V4546L$F+R#e*_b1CU3Z((!c!233#J,s"3399%*!5!5($S%*#J S! s,,c a1P1P  %s3AS2,b r;T)preserve_cpu_scalar_tensorsc^^^^[SU56mS[S[S[4SjmUUU4Sjm[UU4SjU55$)Nc3h# UH(n[U[5(a UROSv M* g7frJ)r?r'rZrMrs r9rO#_maybe_broadcast..s% G$QZ:..!''D 8$s02rrrsreturnc6SSKJnJnJn [ 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)rEsym_andsym_orTrVcg)NzCsizes assumed to be the same due to unbacked broadcasting semanticsrRrRr;r9rT9_maybe_broadcast..should_expand..s]r;F)rarEr|r}rYziprbrm)rrrsrEr|r}rNys r9 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_tensorrZrristrr6)rNrZr_rurs r9__maybe_broadcast+_maybe_broadcast..__maybe_broadcasts 9 6 " "H : & &*u/I/I!/L/LQWWl33xx --H5DG DsJ r;c36># UHnT"UT5v M g7frJrR)rMrNrr_s r9rOry sBT"1l33Ts)rtr%boolrg)ruargsrr_rs` @@@r9_maybe_broadcastrsH$ G$ GLyT@$ BTB BBr;)register_decompositionF)aten_op extra_meta exact_dtyperzc0^^^^S[4UUUU4SjjnU$)Nprimc>^[T5[TS9[[STS9S[S[4UU4Sjj5555nT[ La![ R"TTR5mTb[T5"U5 U$)Nrrrtype_promoting_argstype_promotion_kindrrrzcD>TbT"U5 T"U5n[U/U5$rJ)rC)rrrBrrs r9_ref>_make_elementwise_unary_reference..inner.._ref&s*%1 !WF/V< .inner#s t  - ) + & 3  =N =~ =   * .  = m #''dmmT"U0UD6$rJrR)rkwargsfns r9_fn_make_alias.._fnFs4"6""r;r)rinspect currentframef_back f_globals __module__)rnamers` r9 _make_aliasr>s5#L))+22<T"U/UQ7SU0UD6$)NoutrR)rrrrrs r9r_make_inplace.._fnUs!,d,,V,,r;_r) getmodule) r rrgetattratenrr__all__append)rr inplace_namer_alls` r9 _make_inplacerNss 2Y--kk]!$LL |!< =c BC" R= D L! Jr;rc.[R"U5$rJ)primsr1rs r9r1r1gs 99Q<r;c.[R"U5$rJ)rr2rs r9r2r2o ::a=r;c.[R"U5$rJ)rr3rs r9r3r3t ;;q>r;c.[R"U5$rJ)rr5rs r9r5r5yrr;c.[R"U5$rJ)rr4rs r9r4r4~rr;c.[R"U5$rJ)rr6rs r9r6r6rr;c.[R"U5$rJ)rr7rs r9r7r7rr;c.[R"U5$rJ)rr8rs r9r8r8s   Q r;c.[R"U5$rJ)rr9rs r9r9r9 ::a=r;inputcB[R"UR5$rJ)ris_complex_dtypedtypers r9r/r/s  ! !%++ ..r;c|[R"UR5(dU$[R"U5$rJ)rrrrr:rs r9r:r:s/  ! !%++ . .   u %%r;c.[R"U5$rJ)rr;rs r9r;r; 99Q<r;c.[R"U5$rJ)rr<rs r9r<r<rr;c.[R"U5$rJ)rr?rs r9r?r? == r;c.[R"U5$rJ)rr@rs r9r@r@rr;c.[R"U5$rJ)rerf_invrs r9rArArr;c.[R"U5$rJ)rrBrs r9rBrBrr;c.[R"U5$rJ)rrCrs r9rCrCrr;c.[R"U5$rJ)rrDrs r9rDrDrr;c.[R"U5$rJ)rrFrs r9rFrFrr;rrrrvaluec[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'AssertionErrorr6r!rrrr rnrrG)rrr python_typemsgs r9rGrGs a $ $9$q'CDD eV $ $9$u+GHH%%agg.K  & &tE{K @ @'U }4ST_S``abo ::a r;c^[R"X5n[R"X5 U$rJ)rrGr)rrrrs r9rHrHs! 1A MM! Hr;c.[R"U5$rJ)rbr+rs r9r)r)s   E ""r;c.[R"U5$rJ)rrIrs r9rIrI ;;q>r;rNc[R"[R"[R"U55[R"U55n[R "X5$rJ)rbrrIr1rlr)rNtrunc_xs r9rJrJs< ii EIIaL15::a=AG 99Q  r;c[U[5(d[S[U535e[R "[ R"UR5S5 [R"U5$)Nrcg)Nz#imag only supports complex tensors.rRrRr;r9rTimag.. s1Vr;) r?r'rr6rbrmrrrrrrs r9rrsW a $ $9$q'CDD LL qww')V ::a=r;)rc[R"UR5(d%[R"UR5(a[R "U5$[ U[RS9$)Nr) ris_float_dtyperrrrRr&rbrrs r9rRrRsM  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) rrrrbrrSrrrr1floatr+rrs r9rSrSs 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 rrsr9rTisposinf..+I!''Sr;rr rbrmrrrrrr+rrs`r9rTrT$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 rrsr9rTisneginf..9rr;z-infrrrs`r9rUrU2sf  LL  " "177 ++S AGG$$E&M!!   AUZZ 00r;c.[R"X5$rJrrrs r9rVrV@s 88A>r;rac[R"UR5(a[R"U5S:H$[R "U[R S9$)Nrr)rrrrbrr&rrs r9rWrWIsA  agg&&zz!}!! ??1EJJ //r;c.[R"U5$rJ)r bessel_i0rs r9rXrXTs ??1 r;c.[R"U5$rJ)rrZrs r9rZrZ[s <<?r;c.[R"U5$rJ)rr[rs r9r[r[`rr;c.[R"U5$rJ)rr\rs r9r\r\err;c.[R"U5$rJ)rr]rs r9r]r]jrr;c.[R"U5$rJ)rr^rs r9r^r^orr;dimrc U=(d URn[R"U5n[X5n[U[ XQSS9- U5$)NTkeepdim)rrget_computation_dtyper+r)rrrr result_dtypecomputation_dtypea_s r9r`r`usF #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?rnumelrbrrCr[rrrr1rrrx)rrr maxesmaxes_squeezedresults r9rrs 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'rr6ris_boolean_dtyperis_integer_dtyperrbfinforjminrrVrUrT)rrrrrrs r9rdrds 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.rRrRr;r9rT_neg_meta..s r;)rbrmrrrs r9 _neg_metar#s# LL uzz! r;)rc.[R"U5$rJ)rrers r9reres 99Q<r;c[U[5(d[S[U535eUR[ R La Sn[U5eU$)Nrz'positive does not support bool tensors.)r?r'rr6rrbrri)rrrs r9rfrfsK a $ $9$q'CDDww%**73 Hr;c[U[5(d[S[U535e[R "UR 5(a[R"U5$U$)Nr) r?r'rr6rrrrrrs r9rrsN a $ $9$q'CDD agg&&zz!} Hr;c.[R"U5$rJ)rrhrs r9rhrhs   A r;)decimalsr(cUS:Xa[R"U5$SU-nSU*-n[R"[R"[R"X55U5$)Nr )rrir)rrr(ten_pow ten_neg_pows r9ririsQ1}{{1~h,hY' yyUYYq%:;[IIr;c.[R"U5$rJ)rrrs r9rrrr;c R[S[S[[U5555$NrV)rrxrCrers r9rjrjs q#aSV- ..r;c[R"UR5(a-UR5n[R "US:HSX- 5$UR 5$Nr)rrrr1rbrrl)rra_abss r9rkrksH  agg&&{{5A:q!)44vvxr;c.[R"U5$rJ)rrlrs r9rlrlrr;c.[R"U5$rJ)rrmrs r9rmrm s == r;c.[R"U5$rJ)rrnrs r9rnrnrr;c[RU-n[R"US:HS[R"U5U- 5$NrrV)mathpirbrrnrs r9roros3 ! A ;;qAvq%))A,"2 33r;c.[R"U5$rJ)rrprs r9rprp"rr;c.[R"U5$rJ)rrrrs r9rrrr'rr;c[X5$rJ)rrs r9rsrs,s q9r;c.[R"U5$rJ)rrtrs r9rtrt4rr;c.[R"U5$rJ)rrurs r9ruru9rr;c.[R"U5$rJ)rrwrs r9rwrw>rr;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: rR) input_dtypesr9rT!view_as_complex..Ks55@MCr;rcg)Nz-Input tensor must have one or more dimensionsrRrRr;r9rTrCQ?r;r\cg)Nz+Tensor must have a last dimension of size 2rRrRr;r9rTrCU=r;rVcg)Nz/Tensor must have a last dimension with stride 1rRrRr;r9rTrC[Ar;c30# UH oS-S:Hv M g7f)rFrNrR)rMstrides r9rO"view_as_complex.._s84aZ1_4scg)NzCTensor must have a stride divisible by 2 for all but last dimensionrRrRr;r9rTrC`sUr;cg)Nz0Tensor must have a storage_offset divisible by 2rRrRr;r9rTrCdsBr;)rrbrmrrsizerYrLbuiltinsrstorage_offsetrview_element_typecorresponding_complex_dtyper)rsizes old_stridesdimsrBs @r9rrGs**K LL [) C IIKE LL E a? LL b Q= ++-K LLB1A s D LL 8488U LL !Q&B  " " e// <  gbkr;c @^^^^^^^^S[4UUUUUUUU4SjjnU$)Nrc>^Tc TRm[T5[ST S9S[[[ 4S[[[ 4S[4UUUUU4Sjj55nT(a[ 5"U5nTUlT[La[R"TT5mTbT(a[T5"U5 U$)Nrrrsrrrrsrzc>[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!rRrsr9rTQ_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!rRr]sr9rTr^r_r;c>TS3$)Nz?: Receive two Number inputs to an elementwise binary operation!rRr]sr9rTr^s 4& _`r;)rb _check_valuer?r!rrC)rrrsrBrrsupports_lhs_python_scalarsupports_rhs_python_scalarsupports_two_python_scalarss r9r?_make_elementwise_binary_reference..inner.._refzs   *G*Q2G.G>    *G*Q2G.G>    +I"1f-G*Q2GH`  $A)DA!ZF/? ?r;) rr r.rTensorr"r0rrrr) rrrhas_outrshould_register_decompositionrcrdrers ` r9r1_make_elementwise_binary_reference..innerus <==D t + * 3  @VZ'( @VZ'( @ @ @     @. =&D m #''d3G  #@ "7 +D 1 r;r) rrrrhrcrdrerirs ```````` r9"_make_elementwise_binary_referencerkks*H**X Lr;rZalpharsrmc[X5upUb[U[5(a URO URn[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)rr?r'rrrrr r6rnrrrxrCrrrsrmrrrrBs r9rxrxs A !DA %a44!''))%0 d "5+F+F K, , ,DK=8WXcWddefCS/ ! a $ $ !#A A YYq_F ' 77r;)rrcrdc.[R"X5$rJ)rryrZs r9ryry ;;q r;)rc.[R"X5$rJ)rrzrZs r9rzrz   Q ""r;c.[R"X5$rJ)r shift_leftrZs r9r{r{   A !!r;c.[R"X5$rJ)rr|rZs r9r|r|rwr;c.[R"X5$rJ)rshift_right_arithmeticrZs r9r}r}s  ' ' --r;c.[R"X5$rJ)rr~rZs r9r~r~rtr;)rrcc[U[5(a4[U[5(a[XRUR S9nOl[U[5(aW[U[5(aBUR UR :wa(SUR SUR S3n[ U5e[[U5[[U55[U55$)Nrr8/Expected divisor (b) to be on the same device (&) as dividend (a), but it is found on r) r?r!rgr(rr8rirrmrer1)rrrsrs r9rrs!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 rwrIzLdiv expected rounding_mode to be one of None, 'trunc', or 'floor' but found .)rrrrn)rrrsrrs r9rrsV1  ' !A!! ' !A!!\]j\kklmor;c.[R"X5$rJ)rrrZs r9rr 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!rrbrrrGrrrrrFrrrZs r9rr&s 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!rnrget_higher_dtyperrrb complex128float64r+rrrrrsrs r9rrDs !VAv!6!6 K   " "1 (E }NOO e$$    )A)A A !DA q9r;)rrericZ^[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) Nr}cpur~rrr7z*a and b must both be Tensors at this pointFc>TS3$)Nz not supported for floor_dividerRrsr9rTfloor_divide..s ug-L$Mr;)r?r!r(rgrr8rbrir device_putrrr_floor_divide_floatr_floor_divide_integerrm)rrrsrrs @r9rrs!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- $r1) rr is_signedrrrbrmrrr+)rrrsoffsets r9rrs~ 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$)NrrVrr})rrrr~rrrzrrIrrxr(rr8r) rrrsmodrdifferent_signed_inputsnon_zero_remaindermask floor_div basic_div zero_tensors r9rrs 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$rJ)rrrZs r9rr ::a r;c.[R"X5$rJ)rrrZs r9rrrr;c.[R"X5$rJ)rrrZs r9rrrr;mantissaexponentch[RR[R"U55$rJ)rb return_typesfrexprrs r9rrs$    # #EKK$5 66r;c.[R"X5$rJ)rrrZs r9rrs 99Q?r;c.[R"X5$rJ)rrrZs r9rrrr;c.[R"X5$rJ)rrrZs r9rrrr;valuesc[R"US5n[R"[R"US5[R"U55n[R "USS5n[R "X!U5nU$r7)rbrrrrVr)rr input_eq_zero input_lt_zerozeros_and_onesrBs r9rrsa HHUA&M$$UXXeQ%7U9KLM[[15N [[ ?F Mr;c.[R"X5$rJ)rrrZs r9rrrrr;c.[R"X5$rJ)rrrZs r9rrs << r;c.[R"X5$rJ)rrrZs r9rr"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)rrrsrsr9rT#_check_close_args..4s'4&LQWWIUZ[\[b[bZccder;rc>TSTS3$)Nz6: rtol must be greater than or equal to zero, but got rrR)rrsr9rTr84&NtfTUVr;c>TSTS3$)Nz6: atol must be greater than or equal to zero, but got rrR)rrsr9rTr<rr;)rbrbrrmrrrrsrrs`````r9_check_close_argsr+sV  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)rrrrrrrrrVrconvert_element_typerbget_default_dtyperxr1rrrRr) rrrsrrrclose allowed_error actual_errorrs r9rrAs?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$r7) rrbint8int16rrint32rrr1r)rrrsrpromote_to_intgress r9rrls GGEuzz5;;77N  & &q%++ 6  & &q%++ 6 !A AFAq!A ))EIIaOa' (C$3P%*D*DS*PPr;c.[R"X5$rJ)rrrZs r9rrrr;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)rbrrrrrRrrrr[rCr\rVcomplexr) rrrsrmax_min_inf_mask neg_min_maskinf_vals non_nan_valsnan_masks r9rrs~ ::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 dtypesrRrRr;r9rTlogaddexp2..;r;rrF) rbrmrrrrrrSr8r[r\rF)rrrsrrrr inv_log_2rs r9rrs  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-$r1rrrrZs r9rrG  ! !!'' * * F  ! !!'' * * F 5Lr;c\[R"UR5(dUS:H$U)$r1rrs r9rrs'  ! !!'' * *Av 2Ir;c[R"UR5(dUS:gn[R"UR5(dUS:gn[X5$r1)rrrr|rZs r9rrsK  ! !!'' * * F  ! !!'' * * F a r;c[R"UR5(dUS:gn[R"UR5(dUS:gnX- $r1rrZs r9rrrr;c.[R"X5$rJ)rrrZs r9rrrr;c.[R"X5$rJ)rrrZs r9rr == r;c.[R"X5$rJ)rrrZs r9rrrr;)rrec.[R"X5$rJ)rrrZs r9rrs 99Q?r;c.[R"X5$rJrrZs r9rrrr;c.[R"X5$rJ)rrrZs r9rr s ??1  r;c.[R"X5$rJ)rrrZs r9rrs ??1  r;rVcp[U[5(a Sn[U5e[R"XUS9$)Nz?Received a Number for the first argument, but expected a Tensorrl)r?r!rnrbr)rrrsrmrs r9rrs1!VOo 99Q ''r;c[X5up[U[5(a|[U[5(ag[R"[ R "UR5(+=(a% [ R "UR5(+S5 US:wa[U[5(a URO URn[ 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.rRrRr;r9rTsub..?sCr;rVrorr)rr?r'rbrmrrrrr r6rnrgrrrrCrps r9rr*s A !DA!Z  Z:%>%> &&qww/ / W8N8Nqww8W4W   z%a44!''))%0 **4; DD+DK=8WXcWddefCS/ ! a & & !#A  A YYq_F ' 77r;r)rrrrec.[R"X5$rJ)rrrZs r9rrWs 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"rRrRr;r9rTxlogy..jrJr;r}rzb must be TensorLike, got rr)rbrmr?r'r!r(rr8rr6rrrr[rVr)rrrsrhss r9rras 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;)rrrec[R"U5n[R"U5(a[R"X5$[ [R"X55$rJ)r get_dtyperrrrwrs r9rr~sD 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$Nrrrr5rrsr9rTaddcdiv..-d5k]:YZeYffghr;rrrrbrbr r6rrrrrrs ` @r9rr_   ))%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$rr5rsr9rTaddcmul..rr;rrs ` @r9rrrr;)rrrrjrrjcUcUc 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!)rnrbrVr|rrr)rrrrjra_isnan conditions r9rrs {s{;o ++a.$$UXXa%5w? KK c * ++a.$$UXXa%5w? KK c * Hr;c*[R"XS9$)N)rrbr)rrs r9rr ;;t %%r;c*[R"XS9$)N)rjr)rrjs r9rrrr;predc^UbUc[e[R"TXSS9 [R"TR [R LU4Sj5 [TX5ump[R"TX5$)Tallow_cpu_scalar_tensorsc">STR3$)Nz#expected predicate to be bool, got r)r sr9rTwhere..s5djj\Br;) NotImplementedErrorrcheck_same_devicerbrmrrrrr)r rrrss` r9rrsh" yAI!! D!F LL ejj B "$-JD! ;;tQ ""r; memory_formatrc.[R"XS9nU$Nr)rr)rrrrs r9rr"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!)r8rirr)rrrsrrs r9rr+sK !((ahh"6/zQXXJNxy3 == r;cUR5S:waSUR5S3n[U5e[R"UR5nU"[ R "U55$)NrVzCan't convert a tensor with z elements to a number!)rrnrrrrr)rrr number_types r9rr3sWwwyA~,QWWYK7MNo%%agg.K uzz!} %%r;r8copylayout 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)r8rrrbpreserve_formatris_contiguous_for_memory_format)rrr8rrrrrrs r9_to_will_aliasr"@s  t^ 1qxx61 d] .agg. t^ 1qxx61  T ! U 5 55 U44QT r;c[erJ)r)rrs r9 _to_dispatchr$Ys r;cUUUUUS.nU$N)r8rrrrrRr8rrrrrs r9 _to_devicer(^s!$& F Mr;c>[R"U5UUUUS.nU$r&)rbr8r's r9_to_device_strr*ps*,,v&$& F Mr;cUUUUS.nU$)N)rrrrrR)rrrrrs r9 _to_dtyper,s$& F Mr;otherc`URnURnURnUUUUUUS.nU$)N)r8rrrrr)r8rr)r-rrrr8rrrs r9 _to_otherr/sA\\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)rr8rrr8r) r?rrbr8r r6indexrpop)rrr0options_to_checkkws r9_canonicalize_to_argumentsr6sE9Ih,?!E!E#ll9X+>? ( ?&9=E# UHnTRUS:v M g7frNrZrMirrs r9rO_reduction.. s.Lt!qwwqzA~ts"cg)NzJreducing over zero-size dimension for reduction operation without identityrRrRr;r9rT_reduction.. s`r;rVz4result_dtype should not be None when out is providedz7Expected the dtype of reduction result and out to match copy_fromr)r?r'rr6ndimrirrrreduction_dimsrZrar|rbrmreduction_dtypesr+rlrbroadcast_in_dimr,r-r)rrrr9r:rWr;rrr<r| valid_shaperr rrD output_shapebroadcast_dimss` r9 _reductionrQsA 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)rrbcontiguous_formatrSs r9rT3_make_copy_from_view.._fn..C saggE,C,CgDr;)pytreetree_map)rrrrrreturn_none_on_out_variants r9r!_make_copy_from_view.._fn: sBT-s-f- %#/ ?M D   r;_copy)rrrr0r rSupdater)rr[aten_fn annotationsr copy_names`` r9_make_copy_from_viewrb0 sdBKK(G"/4K  &w /B 2Y     ;;-u%IL{+7434S9 Jr;c[R"[R"[R"U5XS95nUR[R:XaUR [RS9nU$)Nr r)rbrrruint8r)rrrr rs r9rrN sS  uyy):):1)=sT UFww%++- Mr;cp[U[R5n[U[[ 45(a [ U5S:XaUR5nOURXS9RS5nUR[RLa%[R"U[R5$U$)Nr)rr F)r+rbrr?listrgrYrrrrrdrr)rrrr rrs r9rr] 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$NrRrWr;rrr<) rrrrrbint64rQrrr$SAMErrrr rrs r9rrq 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 ""rBrrrZsr9rTsum_to_size.. s%eW,I!''RSTr;rc3n># UH*nTUT- S:XdMTRUS:wdM&Uv M, g7frVNrB)rMrDrr leading_dimsrZs r9rOsum_to_size.. s@50A \! "a ' ,-GGAJ!O 0s 55 5T)rr r)rextract_shape_from_varargsrbrmis_expandable_torZ is_same_shaperYrview_ofrJrgrlr)rrrZ reduce_dimsrvs`` @r9rr 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$rh) rrrrrbrjrQrrr$rkrls r9rr s } ?IIE  # #AGG , ,0F0Fqww0O0OKKEGGE byC2I    499 r;rTc vUS:XdU/:XaSn[U[RUUSUS[RS9$NrRFrWr;rrr9r<)rQrrr$rkrrrr rs r9rr E byC2I    499  r;c vUS:XdU/:XaSn[U[RUUSUS[RS9$r)rQrrr$rkrs r9rr rr;c@Uc[U[5(aUnSnX4$rJ)r?r)runbiaseds r9_dim_var_dispatchr s(JsD11 =r; correctionrrc [X5up[R"X$5nUS:XdU/:XaSn[U[ [ R US9UUSSS[RS9nU$)NrRrTr) rrset_correctionrQrrrr$COMPLEX_TO_FLOAT)rrrrr rrs r9rr 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) rrrrLr$rr+rbrrrr) rrrrr r opmath_dtypera_vara_stds r9rr 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$) NrRric>STcSOSST3$)Nz&mean(): could not infer output dtype. Inputr z> dtype must be either a floating point or complex dtype. Got: rR)r orig_dtypesr9rTmean..L s"4$,w*=>77# UHnTRUv M g7frJrBrCs r9rOmean..U s7QDq Dsr?rH)rrQrrr$KEEP_PROMOTED_TYPErbrmrrrr?rrKrZrJroperatorrrr+r'rr6r,r-) rrrr rrrrWnelemr rs ` ` @r9rr2 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) rrrrLr$rrr+rbrrrr) rrrrr rrroriginal_dtypera_meanrs r9rra s&c4MC%%h;J00 % 6 6LWWN0ANN1jRME JJu E }NOO-7 r;cP[X5up[XX#US9n[XU5nXV4$)Nr)rrr)rrrrr rvms r9rr| s1&c4MC AH*=A QWA 4Kr;)rvec1vec2)betarmrrrc^^^^^^[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--$) NrVc$>STRS3$)Nz*addr: Expected 1-D argument vec1, but got -DrJ)rsr9rTaddr.. STRS3$)Nz*addr: Expected 1-D argument vec2, but got rr)rsr9rTr rr;rmrc>STS3$)NzBoolean z$ only supported for Boolean results.rR)arg_namesr9rTr s(8*,PQr;rc >S[T53$)Nzexpected bool/int beta but got r5)rsr9rTr s5d4j\Br;c >S[T53$)Nz expected bool/int alpha but got r5rlsr9rTr s6tE{mDr;Fc:>S[T5STR3$Nzcannot safely convert z to r6r)rrsr9rTr s,T$ZLTZZLIr;c:>S[T5STR3$rr)rmrsr9rTr s,T%[Mdjj\Jr;)rbrmrJr?rrrrrrZr r6intouterrrr)rrrrrmargrs````` @r9rr 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;rr.c,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)rVrN)rJrrMrrs r9rOatleast_1d.. s#Eu!VVq[i1o5us*,rVr)r? collectionsabcrrrgrY)rrargs_rs r9rr 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'rr6r)rrrarg_s r9_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)rFNr)rMrrunsqueeze_atleast_1ds r9rOatleast_2d.. $Mu!VVq[&:1&==u&)rV) r?rrrrrrrrgrY)rrrrrs @r9rr 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)rMrrunsqueeze_atleast_2ds r9rOatleast_3d.. rrrVr) r?rrrrrrrrgrY)rrrrrs @r9rr s JsKOO$<$<== c;??33 4 4 !UV V "#5z2F MuM MCc(Q,3*CF*r;rPrLrRcZUbUOUR5n[R"XX$5$rJ)rRrr)rrrPrLrRstorage_offset_ints r9rr s1)4!:J:J:L   AV @@r;srcc@UcSOUn[R"XX#U5$r1)rr)rrrPrLrRrs r9rr s'-4.  # #E>P QQr;c:[R"[U65$rJ)rbSizert)rqs r9rr s ::'0 11r;c[U5S:Xa[US[5(dUSn[[ USS065$)NrVrruF)rYr?rgrfrtensorss r9rr# s? 7|qGAJ!?!?!*  'MuM NNr;c[U5[UR5- n[[U[UR5U-55n[R "XU5$rJ)rYrZrgrlrrM)rrrPstartrWs r9rr, 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_formatrbrWr)inputsformatrfs r9 cat_compute_output_memory_format-cat..cat_compute_output_memory_format9 sgA++A.AE+++!fk...F  > !OP P r;rz3cat expects at least one tensor, but received zero!ztensor must be TensorLike, got rFrErVcD>STRSTRSTS3$)N6Number of dimensions of tensors must match. Expected z-D tensors, but got z-D for tensor number in the listr)examplerDrsr9rTcat..l 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_idxsr9rTr s"#LL>*!!+ L:r;c38# UHoRv M g7frJ) requires_gradrLs r9rOcat.. s$Fg__gr[rrr8rrr)rYrnr?r'rr6rrrarErkrJrbrmrZrrr Exceptionrrr8canonicalize_dim validate_idxrrr)rrrrtensorrErZfilteredrrrrDrrs @@@@r9rr2 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 g7fru)rJrrrLs r9rOcolumn_stack.. s3@G1VVaZQYY 1~66s?ArV)rgrraligned_tensorss r9rr s(@GO  ""r;c[R"UR5(dU$UR(a[R "U5$[ R"U5$rJ)rrr is_sparserbr:rrrs r9rr 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$) NrFrc >S[T53$)Nz1Length of pad must be even but instead it equals rX)rsr9rT!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.rX)l_inprsr9rTr s 225c(;P 'r;rVc3*# UH oS:v M g7frArR)rMps r9rO"constant_pad_nd.. s'3aE3sc F>STTT-STTSTTS-STT-S3 $)NzThe input size z, plus negative padding rrVzG resulted in a negative output size, which is invalid. Check dimension z of your input.rR)rD input_sizesl_diffrpad_idxsr9rTr sEok&1*&=%>>V7|nE#gk"2!34117! OMr;rF)rbrmrYrZrlrrQrrrfrrrrrr8rrrGrr)rrrl_padc_input new_shapenew_dimrrBc_outputrDrrrrs ` @@@@@r9rr 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 operatorrRrRr;r9rTcontiguous.. sRr;r)rbrmr rr)rrrs r9rr s? LL...R 0O ;;q 66r;cr[R"[U5S:S5 [U6n[ US5$)Nrcg)Nz%dstack expects a non-empty TensorListrRrRr;r9rTdstack.., +Rr;rF)rbrmrYrrrs r9rr* 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)rErFrHr}rVcg)Nz3expand: the requested shape has too few dimensions!rRrRr;r9rTexpand..D sEr;r\c>STSTS3$)Nz3expand: attempting to expand a dimension of length  -> rrR)requested_lengthrNsr9rTrj sMaSPTUeTffghr;)rarErFrHr}rbrcrdrerfrYr?rrgrmrZrfrkrvalidate_shaperrMrl)rrrrZrErFrHr}rprshape_r` offset_idxx_hintrequested_hintrrNs @@r9rr1 s%%--CCI  5zQ:eAh99eAh LL E c!''l"E Z#agg, &F %[FAGG$Q\  , *b0 1 1!"F Xa[[X6F-G-G"1!*+;!<Q;>Q#6LLa( LLqAv/145h  LL)Q. /!1F I%N   ! ! 5vs177|f'<=> r;c8URUR5$rJ)rrZrZs r9rrx 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) rnrrrJrZr8r9rIrlrrrg) rrr#rrlength chunk_size full_chunkstail_chunk_sizerDrs r9rr| 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)rrrJr_collapse_view_helper collapse_viewcollapse)rrr)r*r  _new_stridess r9rr 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?rgrfrnrcanonicalize_dimsrJvalidate_no_repeating_dimsrrev)rrrWs r9rr s\ dE " ":dD+A+A<==  " "1664 0D $$T* 99Q r;cPURS:a [S5e[US5$)NrFzInput must be >= 2-d.rVrJrirrs r9rr %vvz233 4=r;cPURS:a [S5e[US5$)NrVzInput must be >= 1-d.rr8rs r9rr r9r;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.rRrRr;r9rTnarrow.. s=r;cg)Nz-narrow() cannot be applied to a 0-dim tensor.rRrRr;r9rTr= s&Ur;cg)Nz&narrow(): length must be non-negative.rRrRr;r9rTr= s&Nr;c>ST*STSTS3$)Nz0start out of range (expected to be in range of [, z ], but got )rR) dim_lengthrsr9rTr= s#BJ;-rR\Q]]hinhoopqr;c>STSTSTS3$)Nzstart (z ) + length (z) exceeds dimension size (z).rR)rCr%rsr9rTr= s'% VH4NzlZ\]r;)r?r'rbrmrrrrrr rrrJrP _check_with IndexErrorrfrZrrLrR)rrrrr%r rCs `` @r9rr 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 FTrrr ) rr3rJr rr+r?r'rr6rbrr) rrrGrHra_acc biased_varrrstdrs r9 _normalizerN s$'' :I33AGG< #A 9E eZ ( (=d5k]KLL~~ udJ ;;z' (D <4 C d?r; dimensionscV[U5Hn[R"X5nM U$rJ)sortedrbr)rNrOrs r9_unsqueeze_multiplerR 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$) NrFc">STR3$)Nz=Expected at least 2 dimensions for input tensor but received rrsr9rT#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 = rB)rrXsr9rTr[ s!V #EKK=0B:, OPr;rFTrJrrV)rbrmrJrr rr+rr3rrr8r6rrrfrlrRr rZr)rrSrTrUrVrWrXrHr input_accrKinput_reshapedrLrrMweight_reshapedwrs bias_reshapedrP unsqueeze_w unsqueeze_brunsqueeze_weightunsqueeze_biass` ` r9rr 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_shapec^^^^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 [T5mTb [T5mTb [T5mTRU- n[[UTR55n[TX5upn Tc TbU T-n OTb TcU T-n OTb TbU T-T-n [U TR5n TRRS;a,[U TR5n [U TR5n XU 4$) Nrsym_eqrVc >S[T5-$)NzzExpected normalized_shape to be at least 1-dimensional, i.e., containing at least one element, but got normalized_shape = )r)rgsr9rT#native_layer_norm..` 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 = rrZ)rgrSsr9rTrlj s3 fll  %%    r;cR>S[TR5-S-[T5-$)NzMExpected bias to be of same shape as normalized_shape, but got bias of shape rnro)rTrgsr9rTrls s2 djj/ %%    r;cp>S[T5-S-[T5-S-[TR5-$)NzGiven normalized_shape=z, expected input with shape z, but got input of size ro)rrgsr9rTrl sI)   ()    % % ekk  r;)rmtia)rarjrYrbrmrZrgrJrrfrlrNr+rr8r6) rrgrSrTrHrjnormalized_ndimaxisrKrrrMs ```` r9rrR s=*+O LL1  LL$G&u5E/FG  LL  Ctzz51A+BC  LL o%  KKo58 9 " #    u E F# $ :: 'D%ejj12N . s5Tr;rcg)Nz'renorm: non-positive norm not supportedrRrRr;r9rTr s Ir;cg)Nz#renorm: maxnorm must be real-valuedrRrRr;r9rTr s2Wr;c>ST3$)Nz,renorm: expected maxnorm to be >= 0 but got rR)r|sr9rTr s LWIVr;rVc>STS3$)Nz/renorm: input needs at least 2 dimensions, got dimensionsrRrsr9rTr sA${Sr;T)r rr gHz>r)rbrmr?rrJrrrfrlr rlinalg vector_normrrrr) rrrr|r|acc_typerbrH norm_factorrJs ` @r9r0r0 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 r7rrsr9rTstft.. s'OPUP\P\~ ^ / 0r;cg)Nz6stft only supports align_to_window for center = False.rRrRr;r9rTr sHr;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.rRrRr;r9rTr scr;cg)Nz:stft expected a tensor of floating point or complex valuesrRrRr;r9rTrsLr;rVrFcg)Nzstft expected a 1D or 2D tensorrRrRr;r9rTrs/Pr;rrc>STST3$)Nzstft expected 0 < n_fft <= z, but got n_fft=rR)r%rsr9rTrs-fX5EeWMr;c>ST3$)Nz0stft expected hop_length > 0 but got hop_length=rR) hop_length_sr9rTrs B;-Pr;c>ST3$)Nz9stft expected 0 < win_length <= n_fft but got win_length=rR) win_length_sr9rTrs KK=Yr;c0>STS3STR3-$)Nz8expected a 1D window tensor of size equal to win_length=rAzbut got window with size rB)rrsr9rTr"s#F{mSU V)&,,8 9r;r}r\) dimensionrPsteporthocg)Nz9Cannot have onesided output if window or input is complexrRrRr;r9rTr=Or;rrb)rbrmr8r/rrrrrJr itertoolsrrZrrr rPr%rrfftrfft transpose_squeeze_ view_as_real)rrrrrrrrrrrreturn_complex_ original_ndim extra_dims pad_amountextended_shapeleftinput_pad_amount complex_fftrbrrr%rs`` ` @@@r9r1r1 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 rr7rsr9rTistft..[s(PQVQ]Q]P^ _ / 0r;rc@>STRSTR3-$rr7rsr9rTrfs! Pofmm_= >r;r\rVrcg)Nz"istft input tensor cannot be emptyrRrRr;r9rTros,Pr;rFrc">STR3$)Nz8istft expected a tensor with 2 or 3 dimensions, but got rrsr9rTrrsJ5::,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 rRfft_sizesr9rTrys_GzRSr;c>SST3-4$)Nrz,to match n_fft when onesided=False, but got rRrsr9rTrs_@ KLr;cg)Nz+istft expected 0 < hop_length <= win_lengthrRrRr;r9rTrrHr;cg)Nz&istft expected 0 < win_length <= n_fftrRrRr;r9rTrs*Rr;cg)NzDInvalid window shape. window has to be 1D and length of `win_length`rRrRr;r9rTrsVr;r}rcg)Nz9cannot have onesided output if window or input is complexrRrRr;r9rTrrr;rcg)Nz:Complex windows are incompatible with return_complex=FalserRrRr;r9rTrPr;)rrr%z%Expected input.size(2) == n_fft, got z != )rrrPrzexpected_output_signal_len (z) != y.size(1) (rBz) != 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)rbrmr8rrrrPrrJrZcorresponding_real_dtyper%rrrrrifftrirfftrr unfold_backwardrrrjrwarningswarn)rrrrrrrrr%rrrn_framesexpected_output_signal_len onesided_ real_dtypewindow_rrrby_tmprwindow_enveloprendrs` ` @r9r2r2Ls 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_scalarrrVc>STSTST3$)Nz%Maximum size for tensor at dimension z is z but size is rR)rmax_sizerPsr9rT*_get_unfold_shape_stride..s7uD -X\W]^r;c>STS3$)NzStep is z but must be > 0rRrsr9rTrs(4& 01r;)rYrrrbrmrfr) rrrrPra_ndim last_striderZstridesrrs `` @@r9_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$) NFrncg)Nzbrepeat: Number of dimensions of repeat dims can not be smaller than number of dimensions of tensorrRrRr;r9rTrepeat.. str;rrVc3.# UH upX-v M g7frJrR)rM padded_size repeat_sizes r9rOrepeat..s(G $K !(GrT)keyreverse)rrxrbrmrYrZrrJrrgrrrr8rrmake_contiguous_strides_forrkrrjrfsortr itemgetterrrr)rr repeat_shapenum_new_dimensions padded_shapedim_size target_shapeurtensor_shapeurtensor_striderenumerated_stride permute_order_sorted_striderepeat_xtensor cloned_resultpermuted_results r9rrs33L5QL 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 rB)rr`sr9rT/_reshape_view_helper_core_alg..sgcU"=bhhsm_Mr;)rJrr split_dimrZr.rrLrnrrbrmrr{) rrrZrr%last_dimaccumrr r1rrr`s @@r9_reshape_view_helper_core_algr As( 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$) NFrnrrVzSTRSTS3$)Nz&Could not reshape a tensor with shape rrrBrrsr9rT&_reshape_view_helper..s8 AYZ_Y``abr;)rrx infer_sizerrrrJrrrr{rYrZrrrbrrrrmr )rrrrZ_ar%dim0dim1 shape_numels` ` r9_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;rZc [U/UQ7SS06$)NrT)rrrs r9rrs  ;E ;d ;;r;c@URUR55$rJ)rrPrr-s r9rrs << %%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 dimensionsrVz`shifts` requiredz*shifts and dimensions must align. shifts: z, dims: zExpected len_dims > 1, got Nr7)rr3rJr?rrrrrYrFrirbrrr rZrrr8rOr) rrrrW len_shiftslen_dims tail_shifts tail_dimsfirst_dim_rolledrrPrr`s r9rrs  " "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;rrVkc[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`.rFz2expected total rotation dims == 2, but got dims = z/expected total dims >= 2, but got total dims = rrVz7expected rotation dims to be different, but got dim0 = z and dim1 = rrr) rYrirJrr3rbrrrrW)rrrrWs r9rrs9  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 rR) entry_shapesr9rT%_check_stack_inputs..BsJ;-Wcdr;)rarjrZrlrYrbrm)rrjrDr"s @r9_check_stack_inputsr$;sK<!*""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 TensorListrV) rYrrrrJr$rfrZinsertrbrr r)rr wrapped_dim result_sizesrrs r9rrFs 7|qCDD((1).qrr;rV)rbrmrYrrJrrs r9rrosQ LLW!#RS '*Oq!#?A&&  ""r;cr[R"[U5S:S5 [U6n[ US5$)Nrcg)Nz%vstack expects a non-empty TensorListrRrRr;r9rTvstack..{rr;)rbrmrYrrrs r9r r yrr;rUc*[R"URU5n[R"[ U5S:gS5 UR [URSU5[U5-[URUS-S5-5$)Nrcg)Nz"unflatten: sizes must be non-emptyrRrRr;r9rTunflatten..s*Nr;rV) rrrJrbrmrYr rgrZ)rrrrUs r9rrso   -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 dimensionsrRrRr;r9rTunbind..sGr;rRc3R># UHn[R"UT5v M g7frJ)rbr)rMsrs r9rOunbind..s# +SaEMM!S ! !+Ss$') rrrJrb _check_indexrYrZrgr)rrs `r9rrs~   -C  AGG qG wws|q +0+=+=as+S   r;r2rc\UR[RS9RXU5$r)rrbrWrN)rNrr2rs r9rMrMs+ 77!8!87 9 E E F r;c^[R"URU5n[R"TRS:*U4Sj5 URS:XaUR S5OUn[ S54U-T4-nX4U'U$)NrVc$>STRS3$Nz(Index should have dimension 1 or 0 (got rBrr2sr9rTindex_copy_..:5::,aHr;r)rr3rJrbrmrslice)rNrr2rrr`s ` r9rNrNsq  ! !!&&# .C LL aH &&A+ A1A ;.3 % )C cF Hr;c[XX#SS9$)NFinplace _index_fillrNrr2rs r9rPrPs quU ;;r;c[XX#SS9$)NTrErGrIs r9rQrQs quT ::r;rFcX^^[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("U5n U R+U 5 U n U $)NrVc$>STRS3$r?rr@sr9rT_index_fill..rBr;rc$>STRS3$)NzSTRS3$r?rr@sr9rTindex_select..rBr;r) rr3rJrbrmrrrMrrCr)rNrr2r`s ` r9rOrOs  ! !!&&# .C LL aH zzQ"vv{"--a E8JKK ;.3 % )C 6   r;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 g7frurR)rMr`rPs r9rOsqueeze..sJ*# UH&nT"TRUS:H5(dM"Uv M( g7frurB)rMdrrrEs r9rOrY(s&DDqN1771:?$CDs!1 1rVT)r)rarErgrkrZrrr{rJrr3r?rrYrrfrQ)rrrrWrJ dims_listrDrEs` @r9rrs*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 entriesrRrRr;r9rT"split_with_sizes..=sYr;cV>S[R"T5STRT3$)NzSplit sizes add up to z but got the tensor's size of )rQrrZ)rrr]sr9rTr`Bs,(k)B(CCabfblblmpbqarsr;)rlrYrbrmrErnrQrrZrRrfrrrL)rr]rrDsplitsr split_sizer s``` r9rr4s 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 rVrzntensor_split: non-scalar indices_or_sections tensors must have only one dimension, but received a tensor with r)rrrJrnr?r'r8r6rrblongrr!rrZr8rIrlrrgrrtolistrfrY)rrrdr_dimrsectionsrmin_split_sizenum_splits_one_extrar] split_idxrcindicesrDs r9rrTs|  ! !!&&# .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!rrJrsr9rThsplit.. ^!&&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 rro)rrrrcsr9rTrssSBc(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 r5rdsr9rTrs  !456 8r; rbrmrJr?rrZr _check_typerfrg)rrrdr]rrcs`` @@r9rrs 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$)NrFc:>S[TR5-S-$)NzPtorch.vsplit requires a tensor with at least 2 dimension, but got a tensor with rqrrrsr9rTvsplit..rtr;rc0>STRSSTS3$)NzQtorch.vsplit attempted to split along dimension 0, but the size of the dimension rrvrrB)rrrcsr9rTr~s%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 r5rxsr9rTr~ryr;rz)rrrdr]rcs`` @r9r r s LL !  %w//(  1_ =j!8A!=  Az1-- &u 6 &K ; **r;rc^UR5m[R"TS;U4Sj5 TS:Xa[R"X5$[R"X5$)N)rVrFc>STS3$)Nz'diag(): Supports 1D or 2D tensors. Got DrRrsr9rTdiag..s"I$q Qr;rV)rrbrmrr)rrrJs @r9rrsQ 88:D LL Q qy--""400r;rdim2c `^^SSKJnJn [R"U5nUR X#U5m[ R"TRTR:H=(a5 U"U"S[TRTR5565(+UU4Sj5 [TT5 U$)Nr)rEr}c3.# UH upX:gv M g7frJrR)rMd1d2s r9rO#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 rB)rrsr9rT"diagonal_scatter..s"yyk24::,@r;) rarEr}rclone_preserve_stridesrrbrmrJrrZr) rrrrrrEr}rrs ` @r9rrsM  & &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 )r`rankc>STST3$Nz(diagonal dimensions cannot be identical rArRrrsr9rTdiagonal..* HbQUPVWr;r)rPrLrR) rrrrbrmrRrrrPrLrkrr) rrrrnum_dimsrR diag_sizerDr9rUrrs `` r9rrsxxzH  ! !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 rV)rr`c>STST3$rrRrsr9rTdiag_embed..Yrr;r\rFrr8rrfr8r)rJrrrbrmrPrfrZrQr1r*rr8rrrrrjrlrYr mask_tensorr)rrrrrrt_shapezpaira_rangeb_rangecondrD cond_shapes `` r9rrBs 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 rVr\c3># UHoRSv M g7frurB)rMrs r9rO'_block_diag_iterable..sBzVazsrrFc2>STSTR5S3$)Nz5Input tensors must have 2 or fewer dimensions. Input z has rrf)rDrsr9rT&_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 rr7)r8rDrsr9rTrs&$$*8;qc VWYr;rrf) rr rQrr8rkrbrmrZr*rr) rr tensors_2dncolsr col_startrowcolrrightr8rDs ` @@r9_block_diag_iterabler|sLKRJQjjla/ 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 r9rrs  ((r;rjcURS: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 rqrrFz3torch.dsplit attempted to split along dimension 2, zbut the size of the dimension rvr)rJrir?rrZr)rrrjs r9rrsvvz^_`_e_e^ffr s  (G$$(a-1771:;PTU;U A.qwwqzl:^_g^hhij k   Q ''r;cURS:a[SURS35e[R"USURS:aS5$S5$)NrFz7t() expects a tensor with <= 2 dimensions, but self is rrrV)rJrirbrrs r9rrsS vvzEaffXQ O   ??1aaffqj 88a 88r;cl[R"URS;S5 UR5$)N)rrFcg)NzcThe use of `x.T` on tensors of dimension other than 0 or 2 to reverse their shape is not supported.rRrRr;r9rTT..s 7r;)rbrmrJrrs r9rrs, LL &  335Lr;c.[R"U5$rJ)rr{rs r9rrrr;rc*[R"URX45up4URS::dX:Xa[RR U5$[ [UR55nXEU'X5U'[R"X5$r/) rr3rJrrdefaultrfrlrbr)rrrr_dim0_dim1r{s r9rrsq**166D<@LEvv{dlzz!!!$$aff &L == ))r;ct[URUR5XU5upEURXE5$rJ)rrZrLr)rrrPrrZrs r9rrs4. DKKM9DNE ??5 **r;c\URXU5R[RS9$r)rrrbrW)rrrPrs r9rrs0 ;;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)rrrrVr7r\) rJrrrrbrrZr8rlr) rwinitrrrrrrJrgrrmasked_as r9_cumsumprod_commonrs 66D   +C qyAKKN@@ C!GA aggcl188 4B <<?b D 4:> "~~b!#{{4D)H  88r;c "[[SXX#S9$)Nrrwrrrrrr)rrrrrrrs r9rrs 3Q!E SSr;c "[[SXX#S9$)NrVr)rrrs r9rr$s 4a1U TTr;cxURS-n[R"X!5n[R"X4US9$)NrVr)rJrrr expand_dims)rrrrJs r9rr0s6 66A:D   +C   QT 22r;c0SSKJnJn [R"[ [ [[[4U5SS9n[RRRR(dU"UU5(aU"X5$[U/UQ7SS06$)Nr)_view_has_unbacked_input_view_unbacked_metaFrnr)torch._subclasses.fake_implsrrrrxr rr%rgrbrcrdrerfr)rrrZrr shape_tuples r9r r =s22 U9eI.. /7%K 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)rrrnsr9rT take_along_dim..ds% --.VVH4O||n3 5r;c$>STRS3$)Nz?torch.take_along_dim(): dtype of indices should be int but got z insteadr)rnsr9rTrms M}}oX 'r;r\r) rbrmrJrrrgatherr rfrZrPinfer_size_shapesr) rrrnr self_sizesbroadcast_shapeindices_broadcast indices_sizesself_broadcastrs `` r9rr^s' LL ',,  LL w}}-  {||AFF2J7<<+;<<!''] !,,s+ 11*gllnM(/BW]]+ VVC[ 11-J%a9"&&s+-8||N1BCCr;rrr8rrrrc ^[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 supportedrRrRr;r9rTempty..sJr;c>STS3$)Nz/torch.empty: received an unknown memory format rrRrsr9rTrsEm_TUVr;rrr8rr) rbrmr rrxrWrchannels_last_3d!make_channels_last_3d_strides_for channels_last!make_channels_last_2d_strides_forr)rrr8rrrrZrs ` r9rrs LL...J  , ,U 3E///33E: %00 099%@ U00 0 V 99%@    # r;c2[R"UUUUUS9$)Nr)rr)rZphysical_layoutrrr8rrs r9rrs&    #  r;rrr8rcUc UROUnUc UROUnUc UROUn[R"UUUUUS9$)Nrr8rr)rrr8rbr)rrrPrrr8rs r9r r sQ}AGG%EQXXVFQXXVF ;;   r;c Uc UROUnUc UROUnUc UROUn[R"UUUUUUS9$)z< Reference implementation of torch.Tensor.new_empty_strided r)rrr8rbr)rrrPrLrrr8rs r9r!r!sV }AGG%EQXXVFQXXVF      r;rc [R"U5nUc[R"5n[R"UU[R :XaSOSUUUUUS9$NFrrrrxrbrrrrrr8rrrPs r9r*r*sZ  + +D 1D }'') :: %**$!# r;c Uc UROUnUc UROUnUc UROUn[R"UU=(d UR[R :XaSOSUUUUUS9$rrrr8rbrrrrrPrrr8rrs r9r$r$sm}AGG%EQXXVFQXXVF :: "177uzz1q# r;c [R"U5nUc[R"5n[R"UU[R :XaSOSUUUUUS9$NTrVrrrs r9r%r%.sZ  + +D 1D }'') :: ## r;c Uc UROUnUc UROUnUc UROUn[R"UU=(d UR[R :XaSOSUUUUUS9$rrrs r9r#r#Hsm}AGG%EQXXVFQXXVF :: !!''ejj0a# r; fill_valuec Uc UROUnUc UROUnUc UROUn[R"UUUUUUS9$)Nr)rrr8rbr)rrrPrrrr8rs r9r"r"csT}AGG%EQXXVFQXXVF ::   r;cU$rJrR)rPrrs r9 empty_outr}s Jr;)rr8rrrrc PUc UROUnUc UROUnUc UROUnU[R:wa$[R "UR UUUUUUS9$[R"U5upx[R"UR UUUUUUS9$)Nrr) rrr8rbr rrZr3compute_elementwise_output_logical_to_physical_permr) rrrr8rrrrlogical_to_physical_permrs r9rrs}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 nonzerorRrRr;r9rTarange..s$:r;cgNz7upper bound and lower bound inconsistent with step signrRrRr;r9rTrMr;cgrrRrRr;r9rTrrr;cf[U[5(+=(d [R"U5$rJ)r?rr8rRrSs r9 is_finitearange..is_finites!a!566J$--:JJr;c>STST3$)Nzunsupported range: rrR)rrsr9rTrs%eWD6r;c>ST3$)Nzstep must be finite but got rRrsr9rTrs .tf5r;c3B# UHn[U[5v M g7frJrKrMrs r9rOarange..sIDS 3 8 8D)rrrr8rrVFT)r check_layoutcheck_pin_memoryrbr8device_or_defaultr?rrrmrQrrjrrrrr8r9riotarg get_acc_typer+requires_grad_)rrrrrr8rrrr integer_args is_integerxstartxendxsteprkr%r2rrs``` r9rrs 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;)rrrSc X/n[U[5(aURSU5nOURU5 [U[5(d[ S[ U535eUR5S:n[R"XBS- U5n[R"XAU5nXQU- -U-n[R"[U66nUR5U:wa[R"Xx5n[!X75$)NrRz)weight must be Tensor at this point, got rrV)r?r!r"rrgrr6r1rbrr"compute_elementwise_output_stridesrrLr copy_stridedrC) rrrSrrcoeffbaserBrLs r9rYrYs\F&&!!F+ f ff % %HfWXX ::<3 D KKqj& 1E ;;t% (D Ek "T )F  5 57G7P QF }}& ##F3 ' 77r;)rr8rrrstepsc "^^^^^[T[5(aC[R"TR 5S:HS5 [ T[R 5m[T[5(aC[R"TR 5S:HS5 [ T[R 5m[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 tensorsrRrRr;r9rTlinspace..Irr;cgrrRrRr;r9rTr Orr;c3B# UHn[U[5v M g7frJr?rr s r9rOlinspace..SL8KJsG,,8Kr c>STST3$)Nzlinspace(): inferred dtype z& can't be safely cast to passed dtype rR)default_complex_dtypersr9rTr \s56K5LLrsxryzr;dtype must be torch.dtype, got c>S[T5RS[T5RS[T5RS3$)Nz4received an invalid combination of arguments - got (rArB)r6r)rrrsr9rTr fsBu+r$s),,-RU 0D0D/EQHr;zsteps must be IntLike, got cg)Nz$number of steps must be non-negativerRrRr;r9rTr ks%Kr;)rr8rrrrVrrF)rr)!r?r(rbrmrr+rrQrrrTrrrrr6r{rrropsrrrrrrrjrLr$rkrr)rrrrr8rrrfactory_kwargs empty_tensorr dtype_redrrcast_rgrrr's```` @r9rr9s%(( IIK1  P (u}}=#~&& GGIN P &c5==9||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;rc 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 tensorsrRrRr;r9rTlogspace..Tr;cgr2rRrRr;r9rTr3r4r;c3B# UHn[U[5v M g7frJr#r s r9rOlogspace..r%r zbase must not be complex, got r)rbrrrrr?rrr(rmrr+rQrrTrrrr6rrr) rrrrrr8rrrr'_dtyperets r9rrs }'') {{##E** eY ' 'ENE ~ . . LL q T ,E9E c9 % %#,C ^ , , LL QT *#5C||LU8KLLL % A A  # # %! &$  =d4j\JKK ax!! ..   # C #599T#7 ??r;indexingcgrJrR)rr:s r9rrs r;cgrJrR)r:rs r9rrsr;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$)NrrVz*Expected exactly 1 tensor list/tuple, got c3B# UHn[U[5v M g7frJr?r'rs r9rOmeshgrid..s@1Z:..r cg)Nz)meshgrid expects its inputs to be tensorsrRrRr;r9rTmeshgrid..rr;cg)Nz'meshgrid expects a non-empty TensorListrRrRr;r9rTrBs+Tr;cg)Nz3meshgrid expects all tensors to have the same dtyperRrRr;r9rTrBsIr;cg)Nz4meshgrid expects all tensors to have the same devicerRrRr;r9rTrBsJr;FxyrFijc>ST3$)NzDtorch.meshgrid: indexing must be one of "xy" or "ij", but received: rR)r:sr9rTrBs!!) ,r;zexpected TensorLike, got c>ST3$)NzEtorch.meshgrid: Expected 0D or 1D tensor in the tensor list but got: rR)rsr9rTrBs[\][^_r;r7)r?rfrgrYrrbrmrQrrlrr8r'r6rJrrrkr rrM)r:rrDswap_first_and_second_tensors result_shapegridsrs` @r9rrs^'!*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)rf)rNrMsr9rTmovedim..8s) V ~''+K'8&9 Ar;)rrnc">S[T5S3$)Nz#movedim: repeated dim in `source` (rBrQ)rMsr9rTrRIs5d6l^1Er;c">S[T5S3$)Nz(movedim: repeated dim in `destination` (rBrQ)rNsr9rTrRMs:4 ;L:MQOr;rrV)r6rrbrmrYrJrgrr3setdictrrlr8rr)rrMrNrssdssssdssrrWsidir9rs `` r9rr'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;)rr8rrrrc [R"U5 [R"U5 [R"U5nUc[R "5OUnUc[R "S5OUn[R"UUUUUS9$)Nrr) rr r rxrbrr8rr)rZrrr8rrrs r9rrfsz v :&  , ,U 3E).E # # %EE$*NU\\% F    #  r;valc[R[R[R4H+nU[R"U5R ::dM)Us $ [R $rJ)rbrduint16riinforjrj)r^possible_dtypes r9_strength_reduce_integerrcsI ;; 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 rR)rdsr9rTeye.. #H!Lr;c>ST3$)Nz%m must be greater or equal to 0, got rR)rsr9rTrgrhr;Frr\r7r)rbrmrjr?rIntWithoutSymIntrcrjrrstridedrrr%r) rdrrrr8rr range_dtyperange_nrange_mrones `` r9rrs 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) rr r  type_to_dtyper6rbr8rrG)rZrrrr8rres r9rrs~ v :&&EE,?,?Z@P,QE)Vu||E/BF  #  A ::a $$r;rrr8rrrc L[R"UUUUUUUS9n[X5$)Nrs)rbrrG) rrrrrr8rrrrrs r9rrs6  ## A  r;c [R"UU=(d UR[R:XaSOSUUUUUUS9$)NFrrsrbrrrrrrrr8rrrs r9r+r+sC ?? "177uzz1q##  r;c [R"UU=(d UR[R:XaSOSUUUUUUS9$)NTrVrsrvrws r9r&r&sC ?? !!''ejj0a##  r;c [R"U5 [R"U5n[R"U5n[R"U5n[ R "USSUUUS9$)Nrr)rrrr8r)rr rxdtype_or_defaultrrrc)rr8rrrrZrs r9r'r'se :&  - -e 4F  " "5 )E  $ $V ,F <<  #  r;c[R"U5 [R"U5 UbUO[R"[ U55nUbUO[ R "S5n[R"XUS9$)Nrr}) rr r rqr6rbr8rr()rrrrr8rs r9r(r(6sc 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 r()r|r}rr8rL) rrr?r!rr6rrbrcanonicalize_devicer_uniform_helper)rZr|r}rLrr8s r9rrJs  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 dimensionrR) value_ndimsr9rTmasked_fill..qsVWaVbblmr;cudaxpumpsr4rcg)Nz,Expected `value` to be on same device as `a`rRrRr;r9rTr{sBr;c>STS3$)Nzcould not convert to type z without overflowrR)rsr9rTrs0 =NOr;)rrrr?r!r6rJrbrmr8_C_get_privateuse1_backend_namerr r+rr)rrrr value_type is_cpu_scalarrrrs @@r9rres.%%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$rJ)rbrrO)rrrrrss r9rrs& !5)AGGAJ Hr;c [SXX#S9 [[R"[R"XX#US95R 55$)z, Reference implementation of torch.allclose ztorch.allcloser)rrr)rrrbrrr)rrrsrrrs r9r,r,s@+qDL  %--4iPQVVX r;cR[R"XSS9 [R"X5 URUR:wag[ UR UR 5H up#X#:wdM g UR 5S:Xag[[[X555$)NFr rT) rrcheck_same_dtyperJrrZrrrr)rrrsrNrs r9r-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$)NfrorFr) r?rrYrrgrlrJrbr matrix_normr)rrrr rs r9rbrbs U  z#s';';s3x1}  #se!S ;ejj)*C||''#e'LL||''#e'LLr;c^[R"TRS:HU4Sj5 [R"[R"TS55$)NrFc">STR3$)Nz+expected a matrix, but got tensor with dim rrsr9rTtrace..s=dii[Ir;r)rbrmrJrrrs`r9rvrvs; LL QI 99UZZa( ))r;cp^S[[[4S[[[4S[4U4SjjnU$)Nrrrsrzc>T"X5$rJrR)rrrsbase_ops r9rop_make_r_binary_op..ropsq}r;)rr(r")rrs` r9_make_r_binary_oprs= + , + ,  Jr;rcr[R"URS:S5 URSSup#[R"X0R S9R S5[R"X R S9R S5- U:n[R"X@5R5$)NrFcg)Nz2triu: input tensor must have at least 2 dimensionsrRrRr;r9rTtriu..Qr;rr7r\ rbrmrJrZrr8rrrrrrrhr`rs r9rr LL ! Q 7723. rr;rr7r\rrs r9rrrr;rrcUS:XdUS:XagUS:a[USU-5O[X-S:5n[S[XU-55n[S[XU-55nXC- S-nX4-U-S-nXV- n[SX-5n XyU4$)NrrrrrVrF)rrrj) rrr m_first_row m_last_row n_row_alln_row_trapezoidtrapezoid_sizediff_rowrectangle_sizes r9_get_tril_sizesrs 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 rR)rsr9rT_trilu_checks..7 %CC5#Ir;c>ST3$)Nzcol must be non-negative, got rR)rsr9rTr8rr;c>STSTS3$)Nrqz" not implemented for ''rR)rrsr9rTr;s"TF25';r;)rbrmrrj)rrrrrrs```` r9 _trilu_checksr/sO 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$)Nrrrr8rrrrFrV) rrrjrrbrrrIrrr+rr)rrrrrr8rrrr row_offset arange_kwxs1rs row_inds1 col_inds1xs2 row_inds2 col_inds2s r9rr@s7.#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$)NrrrV)rjrr) rrrrrtrapezoid_size_trilrectangle_size_trilr triu_sizers r9_get_triu_sizesrks 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$) NrrrrgrFrVr) rrrjrrbrrrIrrr+rr)rrrrrr8rrrr col_offsetrrrrrrsrrs r9rr|s:.#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$)NrVc,>STR5S3$)Nz2boundaries tensor must be 1 dimension but got dim(rBrf)rsr9rTbucketize..sDZ^^EUDVVWXr;r\rrrFr)rbrmrr?rgrrrjrZr+r*r8rr&rrr8r]rlr)rrrrr out_dtype n_boundariesrrmidmid_valcond_mid cond_updatenitersrs ` r9r.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 rrsr9rTcauchy..::>**Gr;rc>ST3$)Nz-cauchy_ expects sigma > 0.0, but found sigma=rR)sigmasr9rTrs ?wGr;r) rrbrmrrrrrrtr8r9 rand_like)rmedianr generators` ` r9rrsABB 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 rrsr9rTexponential..rr;rc>ST3$)Nz4exponential_ expects lambda > 0.0, but found lambda=rR)ratesr9rTr s FtfMr;rFrr\) rrbrmrrrrrrrrHrr[)rrr uniform_valepsilonr log_uniforms`` r9rErEsABB 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 rrsr9rTgeometric..)s0 =r;rrVc>ST3$)Nz0geometric_ expects p to be in (0, 1), but got p=rR)rsr9rTr-s B1#Fr;) rrbrmrrrrrIr\rr8)rrrs`` r9rKrKsABB 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 rrsr9rTlog_normal..?s1$**>r;rc>ST3$)Nz-log_normal_ expects std > 0.0, but found std=rRrsr9rTrCs ?uEr;) rrbrmrrrrrrC randn_likerrrrs` ` r9r_r_2sABB LL  " "4:: .. 3&&tzz2 2 3&&tzz2 2>   LL CE 99S5++D11D8 99r;)rr)rrrr8rc  ^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 rRrsr9rTnormal..cs I#Or;c3T# UHn[U[5(dMUv M g7frJr?rxs r9rOnormal..gsL;a*Q 2K;s( (cg)NzFnormal expects that either mean or std is a tensor, or size is definedrRrRr;r9rTrjs\r;cg)Nz.Cannot pass layout, or pin_memory without sizerRrRr;r9rTrnsDr;c38# UHoRv M g7frJrBrxs r9rOrqs"normal expects mean and std to be scalars when size is definedrRrRr;r9rTrwsTr;rrrF)rrrr8rr)rbrkrr?r'rmrgrYrtrr8rrrc) rrrPrrrr8rrnormal_sampless ` r9rcrcJs^(f 5I&RSS c: & & 1HO  |LD#;LL L1  \   dN 1zT1 D !".rEr;gcܥL@rbrmrrr)rM_180_PIs r9rgrgs6 LL  " "4:: ..?CH ?r;c[R"[R"UR5(+S5 SnX-$)Ncg)Nz-deg2rad is not supported for complex tensors.rRrRr;r9rTdeg2rad..rEr;g9RFߑ?r)rM_PI_180s r9r>r>s6 LL  " "4:: ..?EH ?r;c*US:gRU5$r1)r)rrs r9r=r=s AI??3 r;cv^^[R"TR5S:H=(a TR5S:HUU4Sj5 [R"TRTR:HUU4Sj5 UU4Sjn[R"TR 5TR 5:HU5 g)NrVcN>STR5STR5S3$)Nz1D tensors expected, but got zD and z D tensorsrfr-rsr9rT_dot_check..s!/ |6%))+iXr;c<>STRSTR3$)Nz:dot : expected both vectors to have same dtype, but found rrr sr9rTrsL ::,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)rr sr9 numel_error_dot_check..numel_errorsL9$**,{SXS^S^S`Rab004 ~U5;;=/Qg i r;)rbrmrrr)rr-rs`` r9 _dot_checkrsx LL  a,EIIK1,X  LL ekk! *   LL. [X5 T"X5$rJ)r)rr-rs r9wrapper#_dot_check_wrapper..wrappers4$r;)r )rrs` r9_dot_check_wrapperrs  2Y Nr;rcUR5(aUR5(a{UR5(aA[R"UR 5UR 55R 5$[R "UR 5U5$UR5(a%[R "UR 5U5$X-R 5$rJ)r/is_conjrbrrrrrs r9rrs  <<>>}}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$rJ)r/rbrrrrr:rrs r9rrs ??  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$)NrVr\rr7) rrrJrZrbrr8r rrr)rNrrr2 mask_shapers r9select_scatterrs   -CqvvJJO qy $ << QXX 6 ; ;J G5 PD //# # * *177 3C ;;t! $$r;)r[cV[U[R[R45$rJ)r?rb TypedStorageUntypedStorage)objs r9 _isStorager#os 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?rfrgrYitemsizerrnr6rr)seq scalar_typeMAX_DIMS is_storagerUr%handles r9_compute_sizesr,tsHCJ 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?rrbrrrrjrrrcfloatdoublecdoublehalfchalfrirgrr TypeErrorr6rrfrgrYrl_infer_scalar_type promote_types)r" default_dtype scalarTyper%rDcur_itemitem_scalarTypes r9r4r4s#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;r7r"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)NrVrT)rr8rrRrr) r?rgrdetachrr r!rbr(rr_recursive_build)r7r"r'rs r9r<r<s#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 variablerRrRr;r9rT)_internal_new_from_data..s$Rr;F)r8rrr__cuda_array_interface__r8meta)rr)r?rbrgrmr;rr8rhasattrNotImplementedr4r#r6r<) optionsr( device_optdatacopy_variables copy_numpytype_inferencerrinferred_scalar_typer8rs r9_internal_new_from_datarKs$ %% NR  **,C,:syy )53::vv&   t/00&1Zwx7HF7E-d3;$ <<  $ $ .! ! ""6=6eRWX Mr;)rr8rrc $[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)rFrr8rT)rGrHrIr) r?rbrgrr UserWarningrKrdetach_r)rFrr8rrrI new_tensors r9rrs$ %%  l   d]N( 5"(?(?(A % J!!-0 r;rJ)F)rNN)gh㈵>g:0yE>Fr7)NN)NNNNNFF)FFN)NF)NNFr)rr\) NNNTreflectFNNN)NNNTFNNF)rR)rVr)rrrV)rrr\)rNrV)r*)rr)rNF)rrVNru)rVrFN(rQrrrr8rrcollections.abcrrrenumr functoolsrrr r typingr r r rrrb torch._prims_primsrtorch._prims_common _prims_commonrtorch.utils._pytree_pytreerYrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*torch._prims_common.wrappersr+r,r-r.r/r0rrgr DispatchKey_opsr+rr:rCrtr torch._decomprobjectrrrrrr1 INT_TO_FLOATr2r3r5r4r6r7DEFAULTr8r9r/r:r;r<r?r@rArBrCrDrF NO_OPMATHrGrHr)rIrJr ALWAYS_BOOLrRrSrTrUrVspecial multigammalnrarWrXrZr[r\r]r^rrr`rrrdr#rerfrrhrirrjrkrlrmrnrorprr BOOL_TO_LONGrsrtrurwrrkrxryrzr{r|r}r~rrrrrrrrrrrrrrgrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr ScalarSelf ScalarOtherScalarself_outrr rrrrr"r$registerr8rVr(r*r,r/r6rrQrbrrr dim_IntList IntList_outrfrrrrrrrrrrrrrrrrrrrrpy_implCompositeImplicitAutogradMetarrrrrrWrrrrrrrrrrrNrRrr _subclasses fake_implsregister_op_implrxrr0r1rr2rrr rrrrrr$rrqrr rrrMrNrPrQrHrLrOrrWrrrr rrrrrrrrrrrrrrrrrrrr r rrrkrrr r!r*r$r%r#r"rrr start_step start_outrYrrrrrrcrrrr+r&r'r(rrrr,r-rbrvrrrrrrrrrrgrrr.rrErKr_rcrrgr>r=rrrrrabs_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_rrrrrrrrrrrr r#r,r4r<rKrtorch._refs._conversionstorch._refs.ffttorch._refs.linalgtorch._refs.nn.functionaltorch._refs.specialrRr;r9rsQ 88<<77 #$$662l \ hh""  zz~~ 2 Od9=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 EKK X XX' #!7DD :?, ,',26,, ( ,"( !$#'#'  *  Z  Z   )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  ## %"&66t! #"7??#' 8 ^Z '(8 ^Z '(8 J  8 " 8:$7DD$$   $7??#>#n###$7??"."^"""$7??"."^"""$7??.>.n...$7??#>#n###$7DD$ 2 ^Z '( 2-2>:3M-N 2  2 !  $(  ^Z '( ^Z '(C= "($7CC$.^ $7DD ^Z '( ^Z '(6  ^Z '( ^Z '( t$==EE $"' O ^Z '(O-2>:3M-NO  OBFVFF6F161f112$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!~!!  "  ( ^Z '( ( ^Z '( (  (# (! #"7?? $8 ^Z '($8 ^Z '($8  $8 " $8N$7DD   $  >n   # #"7DD:U>:- .:59S3T: $ :.$==EE  $ " ^Z '("-2>:3M-N"  " % #67DD , , ,,  ,  , & ,* % #67?? , , ,,  ,  , & ,* # #+7?? -1,0  ( )  ( )   $  4' -1& & ( )&&(&' -1& & ( )&&(& ( --. ../ ))* ++, #"7AA +/*.# #&'#&'# -+0/)#. # ?D?T?T).)<)<$9=v& "&N&z&#&(,#'%)37!& ^ $ EKK  4.  U\\ "  E//0  237  LL ;;   E//0   #s(^ "37   ;;   E//0   #s(^ "37  ;;    E//0   #s(^  37   E//0   #s(^ ."&"T"0 . n X"##' CC C C  C 8  CC EKK C & C2CCL<! #   (    " ! # (  "$--txx/C/CDE6: $(  x}htCy11 2 EKK  &  F:C C C, "6:   x}htCy11 2 &  #: "# !  (  &    #. "# !  (  &    #.! ##  (,  ( tn  $ "6! 6:# 1 (, 11 x}htCy11 21tn1 1 $ 11"1, "#+  ++ ( +++#+\ & VV# $'+  ( tn    $ '2 & VV##  (,   (  tn   $ '  " #07?? 7A 7A 7A 7A  7A  7A7A # 7Av + ~x77 8 +AO + >5!45 56 +   # *8   + ~x77 8 +AO + >5!45 56 + + ~x77 8 +AO + >5!45 56 +&%) A A  A  ASM A  A//0  %) R R R R  R SM 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 #&#>##  $/4BfBBB%BLFv^Ss5QTAT;U,1~1#1C11* " N"2~#nn(-c>.A(BLOD "). 666 !"@ > tCy ^ ..667D D V D 6 D D  D  DD D 666 !"D8DN../ VVV$H HH V H 6  H  H  666 !" H%0HV..t/E/E/M/MN$O$  %,~,,&, $ #. #.,#.36#.AO#.#.%#.N+??@!% $##%)&*p? p? p? p? p? V  p?  p?p?p?tnp?TNp?d^p? p?Ap?hKAAB!% $## Z Z Z  Z  Z V  Z  Z Z tnZ  SMZ  Z CZ | ",9<DGOR8 $ 51f5151%51pRR*.RRj8?N8?8?8?~<~N8>H8>H8>n8>#8>v # >>)9>>$>< !3   #  G% GC G G$ G" $(  EKK  * #&#>## #&#>##SScS)SS  $ n 3 /A % " *3z:  :  C   J  ( <<<$.<73  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 EKK 9 & 990 $ $( TT T EKK T &  T  T%T % $( UU U EKK U &  U  U&U'33c3n3(3 ))*=N=Ii0@$@=^=+=*#.##N# ^  AE%D ||%D#ll%D19#%D \\%D%DP $( =='+).)@)@% EKK % LL% ^ $ %  %  %&&%%%P $( =='+ EKK  LL  ^ $   $'  $(%)'+  EKK  U\\ "  ^ $ (,../  $(%)'+   EKK  U\\ "  ^ $08 **+ $( =='+  EKK  LL ^ $     ,0'  $(%)'+  EKK  U\\ "  ^ $ (2 ))* $( =='+  EKK  LL ^ $     +0 &  $(%)'+  EKK  U\\ "  ^ $ '2 &  $(%)'+  EKK  U\\ "  ^ $'0 ==>48  E//0 ?( $('+%)).)>)>&& EKK & ^ $ & U\\ " &  &&&&&&)&R//1F1FGH  $c $( =='+c c * c c EKK c LL c ^ $ccccIcL " #27??88V8U6:3E-F8 # 84 &  $('+ ==X/ ^+ ,X/ z>) *X/ X/ EKK X/ ^ $ X/ LLX/X/X/X/'X/v &   9@ $('+ ==9@ ^+ ,9@ z>) *9@ 9@  9@ EKK 9@ ^ $9@ LL9@9@9@9@'9@x  h~. #     ~     &CND$8%:OOPCC .C'CN; ; #'' (;s,,-; ;~**+  $('+ == E),, -  EKK  ^ $  LL ,6#%++! %)$( =='+%) %)}%) EKK %) LL %) ^ $ %)%)%)%)"%)V**DIIMM:;  $( =='+% %% EKK % LL % ^ $ %%%%<%>$(%)'+).)>)> EKK  U\\ "  ^ $ &&.( $(%)'+).)>)> EKK  U\\ "  ^ $   &&),' $(%)'+).)>)> EKK  U\\ "  ^ $   &&(, **+ $('+%)  EKK  ^ $ U\\ "     ,8$( =='+ > > EKK  > LL > ^ $ >  > >,$'$'  tS%  c5 !   ;;  6(() *>**@V***Z))*  + 4J  +         "^4& " &+" M $( M MeSj!"M ( M M EKK 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**+  Xh/ , =*! #)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 "'(>@R(R"SH5rtUL  r;