+ jX<^RIHt^RIHt^RIt.RNRNRNRNRNR NR NR NR NR NRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNR NR!NR"NR#Nt!R$R4t!R%R&]4tR't!R(R)] ]4t !R*R+]4t !R,R]4t !R-R.]4t !R/R0]4t!R1R2]4t!R3R4]4t!R5R6]4t!R7R8]4t!R9R:]4t!R;R<]4t!R=R>]4t!R?R@]4t!RARB]4t!RCRD]4t!RERF]4t!RGRH]4t!RIRJ]4t!RKRL]4t!RMRN]4t!RORP]4t!RQRR]4t!RSRT]4t !RURV]4t!!RWRX]4t"]!4t#] t$] t%] !4t&]!4t']!^4t(]!^4t)]t*]!4t+]%!]+^4t,]!RY4t-]!RY4t.]t/]t0]t1]t2]!RYRZ4t3]t4]t5]!4t6]!4t7]!4t8]!4t9]!4t:]!4t;]!4t<] !4t=]!t>]"t?R#)[)Callable)AnyN Constraintbooleancat corr_cholesky dependentdependent_property greater_thangreater_than_eq independentinteger_intervalintervalhalf_open_interval is_dependent less_thanlower_choleskylower_triangularMixtureSameFamilyConstraint multinomial nonnegativenonnegative_integerone_hotpositivepositive_semidefinitepositive_definitepositive_integerreal real_vectorsimplexsquarestack symmetric unit_intervalc8a]tRt^PtoRtRt^tRtRtRt Vt R#)ra Abstract base class for constraints. A constraint object represents a region over which a variable is valid, e.g. within which a variable can be optimized. Attributes: is_discrete (bool): Whether constrained space is discrete. Defaults to False. event_dim (int): Number of rightmost dimensions that together define an event. The :meth:`check` method will remove this many dimensions when computing validity. Fc \h)z{ Returns a byte tensor of ``sample_shape + batch_shape`` indicating whether each event in value satisfies this constraint. )NotImplementedErrorselfvalues&&`/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/distributions/constraints.pycheckConstraint.checkbs "!cJVPPR,R,#)NNz()) __class____name__r)s&r+__repr__Constraint.__repr__is~~&&r*T11r.N) r3 __module__ __qualname____firstlineno____doc__ is_discrete event_dimr,r5__static_attributes____classdictcell__ __classdict__s@r+rrPs% KI"22r.caa]tRt^mtoRtR]R]/V3Rllt]V3RlRl4t]V3RlRl4t R]R]/R lt R t R t Vt V;t#) _Dependenta! Placeholder for variables whose support depends on other variables. These variables obey no simple coordinate-wise constraints. Args: is_discrete (bool): Optional value of ``.is_discrete`` in case this can be computed statically. If not provided, access to the ``.is_discrete`` attribute will raise a NotImplementedError. event_dim (int): Optional value of ``.event_dim`` in case this can be computed statically. If not provided, access to the ``.event_dim`` attribute will raise a NotImplementedError. r<r=c<<WnW n\SV` 4R#N) _is_discrete _event_dimsuper__init__)r)r<r=r2s&$$r+rI_Dependent.__init__{s'# r.c <V^8dQhRS[/#returnbool)formatrAs"r+ __annotate___Dependent.__annotate__s!!T!r.cXVP\Jd \R4hVP#)z,.is_discrete cannot be determined statically)rFNotImplementedr'r4s&r+r<_Dependent.is_discretes(    .%&TU U   r.c <V^8dQhRS[/#rLint)rQrAs"r+rRrSs3r.cXVP\Jd \R4hVP#)z*.event_dim cannot be determined statically)rGrUr'r4s&r+r=_Dependent.event_dims$ ??n ,%&RS Sr.c rV\Jd VPpV\Jd VPp\WR7#)zo Support for syntax to customize static attributes:: constraints.dependent(is_discrete=True, event_dim=1) r<r=)rUrFrGrC)r)r<r=s&$$r+__call___Dependent.__call__s3 . (++K  &IkGGr.c\R4h)z1Cannot determine validity of dependent constraint) ValueErrorr)xs&&r+r,_Dependent.checksLMMr.rGrF)r3r8r9r:r;rUrIpropertyr<r=r^r,r>r? __classcell__r2rAs@@r+rCrCmsg n !!  Hn H HNNr.rCc"\V\4#)a Checks if ``constraint`` is a ``_Dependent`` object. Args: constraint : A ``Constraint`` object. Returns: ``bool``: True if ``constraint`` can be refined to the type ``_Dependent``, False otherwise. Examples: >>> import torch >>> from torch.distributions import Bernoulli >>> from torch.distributions.constraints import is_dependent >>> dist = Bernoulli(probs=torch.tensor([0.6], requires_grad=True)) >>> constraint1 = dist.arg_constraints["probs"] >>> constraint2 = dist.arg_constraints["logits"] >>> for constraint in [constraint1, constraint2]: >>> if is_dependent(constraint): >>> continue ) isinstancerC) constraints&r+rrs. j* --r.cdaa]tRt^toRtR R]R]/V3RlV3RlllltV3RlRltRtVt V;t #) _DependentPropertya Decorator that extends @property to act like a `Dependent` constraint when called on a class and act like a property when called on an object. Example:: class Uniform(Distribution): def __init__(self, low, high): self.low = low self.high = high @constraints.dependent_property(is_discrete=False, event_dim=0) def support(self): return constraints.interval(self.low, self.high) Args: fn (Callable): The function to be decorated. is_discrete (bool): Optional value of ``.is_discrete`` in case this can be computed statically. If not provided, access to the ``.is_discrete`` attribute will raise a NotImplementedError. event_dim (int): Optional value of ``.event_dim`` in case this can be computed statically. If not provided, access to the ``.event_dim`` attribute will raise a NotImplementedError. r<r=cn<V^8dQhRS[RS[3,R,RS[R,RS[R,RR/#)rMfn.Nr<r=rN)rrrPrY)rQrAs"r+rR_DependentProperty.__annotate__sI $ $ S#X  % $D[ $ : $  $r.c><\SV`V4W nW0nR#rE)rHrIrFrG)r)ror<r=r2s&&$$r+rI_DependentProperty.__init__s '#r.c8<V^8dQhRS[RS[3,RR/#)rMro.rNrm)rr)rQrAs"r+rRrps%   8CH-  2F  r.c D\WPVPR7#)z Support for syntax to customize static attributes:: @constraints.dependent_property(is_discrete=True, event_dim=1) def support(self): ... r])rmrFrG)r)ros&&r+r^_DependentProperty.__call__s " --  r.rerE) r3r8r9r:r;rUrIr^r>r?rgrhs@@r+rmrms72 $$2 $ !/ $ $    r.rmczaa]tRt^toRtV3Rlt]V3RlRl4t]V3RlRl4tRt Rt R t Vt V;t #) _IndependentConstraintz Wraps a constraint by aggregating over ``reinterpreted_batch_ndims``-many dims in :meth:`check`, so that an event is valid only if all its independent entries are valid. cB<\V\4'g"\R\V4P 24h\V\ 4'g"\R\V4P 24hV^8d\RV 24hWnW n\SV`%4R#)*base_constraint must be a Constraint, got z.reinterpreted_batch_ndims must be an int, got z,reinterpreted_batch_ndims must be >= 0, got N) rjrAssertionErrortyper3rYbase_constraintreinterpreted_batch_ndimsrHrI)r)r|r}r2s&&&r+rI_IndependentConstraint.__init__s/:66 ?X>YZ  /)B& r.c <V^8dQhRS[/#rLrO)rQrAs"r+rR#_IndependentConstraint.__annotate__00T0r.c.VPP#rEr|r<r4s&r+r<"_IndependentConstraint.is_discrete##///r.c <V^8dQhRS[/#rLrX)rQrAs"r+rRrsOO3Or.cPVPPVP,#rE)r|r=r}r4s&r+r= _IndependentConstraint.event_dims##--0N0NNNr.cVPPV4pVP4VP8dGVPPVP,p\ RV RVP4 24hVP VPRVP4VP, R,4pVPR4pV#)Expected value.dim() >= but got Nr) r|r,dimr}r=rareshapeshapeall)r)r*resultexpecteds&& r+r,_IndependentConstraint.checks%%++E2 ::<$88 8++558V8VVH*8*Ieiik]K  LLH6::<$*H*HH IE Q B r.cVPPR, R\VP4 RVP R2#)r0(z, ))r2r3reprr|r}r4s&r+r5_IndependentConstraint.__repr__s?..))"-.aT5I5I0J/K2dNlNlMmmnoor.)r|r}r3r8r9r:r;rIrfr<r=r,r5r>r?rgrhs@@r+rwrwsJ "00OO ppr.rwczaa]tRtRtoRtV3Rlt]V3RlRl4t]V3RlRl4tRt R t R t Vt V;t #) riax Constraint for the :class:`~torch.distributions.MixtureSameFamily` distribution that adds back the rightmost batch dimension before performing the validity check with the component distribution constraint. Args: base_constraint: The ``Constraint`` object of the component distribution of the :class:`~torch.distributions.MixtureSameFamily` distribution. c<\V\4'g"\R\V4P 24hWn\ SV`4R#)ryN)rjrrzr{r3r|rHrI)r)r|r2s&&r+rI$MixtureSameFamilyConstraint.__init__%sF/:66 ?%%++,<= 99; '*4>>*:)EIIK=Q  ))+6 -=o > FGB r.c`VPP R\VP4 R2#)rr)r2r3rr|r4s&r+r5$MixtureSameFamilyConstraint.__repr__Es,..))*!D1E1E,F+GqIIr.)r|rrhs@@r+rrsH 00.. JJr.c.a]tRtRtoRtRtRtRtVtR#)_BooleaniIz' Constrain to the two values `{0, 1}`. Tc V^8HV^8H,#rr7r(s&&r+r,_Boolean.checkPs uz**r.r7N) r3r8r9r:r;r<r,r>r?r@s@r+rrIsK++r.rc2a]tRtRtoRtRt^tRtRtVt R#)_OneHotiTz Constrain to one-hot vectors. TcV^8HV^8H,pVPR4P^4pVPR4V,#rr)sumeqr)r)r* is_boolean is_normalizeds&& r+r, _OneHot.check\s@qjUaZ0  " ((+ ~~b!M11r.r7N) r3r8r9r:r;r<r=r,r>r?r@s@r+rrTs KI22r.rcFaa]tRtRtoRtRtV3RltRtRtRt Vt V;t #)_IntegerIntervalibz@ Constrain to an integer interval `[lower_bound, upper_bound]`. Tc<<WnW n\SV` 4R#rE lower_bound upper_boundrHrIr)rrr2s&&&r+rI_IntegerInterval.__init__i&& r.chV^,^8HVPV8*,WP8*,#r1rrr(s&&r+r,_IntegerInterval.checkns0 QY!^ 0 0E 9 :eGWGW>W X r.cVPPR,pVRVP RVP R2, pV#r0 (lower_bound=z, upper_bound=rr2r3rrr) fmt_strings& r+r5_IntegerInterval.__repr__sH^^,,R0 D,,-^Dr?rgrhs@@r+rrbs%K  r.rcFaa]tRtRtoRtRtV3RltRtRtRt Vt V;t #)_IntegerLessThani{z9 Constrain to an integer interval `(-inf, upper_bound]`. Tc0<Wn\SV` 4R#rErrHrIr)rr2s&&r+rI_IntegerLessThan.__init__& r.c@V^,^8HWP8*,#rrr(s&&r+r,_IntegerLessThan.check Q5,<,<#<==r.cnVPPR,pVRVP R2, pV#r0z (upper_bound=rr2r3rrs& r+r5_IntegerLessThan.__repr__6^^,,R0  d&6&6%7q99 r.rrrhs@@r+rr{%K>r.rcFaa]tRtRtoRtRtV3RltRtRtRt Vt V;t #)_IntegerGreaterThaniz8 Constrain to an integer interval `[lower_bound, inf)`. Tc0<Wn\SV` 4R#rErrHrIr)rr2s&&r+rI_IntegerGreaterThan.__init__rr.c@V^,^8HWP8,#rrr(s&&r+r,_IntegerGreaterThan.checkrr.cnVPPR,pVRVP R2, pV#r0rrr2r3rrs& r+r5_IntegerGreaterThan.__repr__rr.rrrhs@@r+rrrr.rc*a]tRtRtoRtRtRtVtR#)_Realiz> Trivially constrain to the extended real line `[-inf, inf]`. c W8H#rEr7r(s&&r+r, _Real.checks ~r.r7N)r3r8r9r:r;r,r>r?r@s@r+rrsr.rcBaa]tRtRtoRtV3RltRtRtRtVt V;t #) _GreaterThaniz5 Constrain to a real half line `(lower_bound, inf]`. c0<Wn\SV` 4R#rErrs&&r+rI_GreaterThan.__init__rr.c VPV8#rErr(s&&r+r,_GreaterThan.checks%''r.cnVPPR,pVRVP R2, pV#rrrs& r+r5_GreaterThan.__repr__rr.r r3r8r9r:r;rIr,r5r>r?rgrhs@@r+rr(r.rcBaa]tRtRtoRtV3RltRtRtRtVt V;t #)_GreaterThanEqiz5 Constrain to a real half line `[lower_bound, inf)`. c0<Wn\SV` 4R#rErrs&&r+rI_GreaterThanEq.__init__rr.c VPV8*#rErr(s&&r+r,_GreaterThanEq.checks5((r.cnVPPR,pVRVP R2, pV#rrrs& r+r5_GreaterThanEq.__repr__rr.rrrhs@@r+rrs)r.rcBaa]tRtRtoRtV3RltRtRtRtVt V;t #) _LessThaniz6 Constrain to a real half line `[-inf, upper_bound)`. c0<Wn\SV` 4R#rErrs&&r+rI_LessThan.__init__rr.cWP8#rErr(s&&r+r,_LessThan.checks''''r.cnVPPR,pVRVP R2, pV#rrrs& r+r5_LessThan.__repr__rr.rrrhs@@r+rrrr.rcBaa]tRtRtoRtV3RltRtRtRtVt V;t #) _Intervaliz< Constrain to a real interval `[lower_bound, upper_bound]`. c<<WnW n\SV` 4R#rErrs&&&r+rI_Interval.__init__rr.cFVPV8*WP8*,#rErr(s&&r+r,_Interval.checks!  E)e7G7G.GHHr.cVPPR,pVRVP RVP R2, pV#rrrs& r+r5_Interval.__repr__rr.rrrhs@@r+r r s Ir.r cBaa]tRtRtoRtV3RltRtRtRtVt V;t #)_HalfOpenIntervaliz< Constrain to a real interval `[lower_bound, upper_bound)`. c<<WnW n\SV` 4R#rErrs&&&r+rI_HalfOpenInterval.__init__rr.cFVPV8*WP8,#rErr(s&&r+r,_HalfOpenInterval.checks!  E)e6F6F.FGGr.cVPPR,pVRVP RVP R2, pV#rrrs& r+r5_HalfOpenInterval.__repr__rr.rrrhs@@r+rrs Hr.rc.a]tRtRtoRt^tRtRtVtR#)_Simplexi zt Constrain to the unit simplex in the innermost (rightmost) dimension. Specifically: `x >= 0` and `x.sum(-1) == 1`. c\P!V^8RR7VPR4^, P4R8,#)rrư>r)torchrrabsr(s&&r+r,_Simplex.checks7yy!,21B0G0G0ID0PQQr.r7N r3r8r9r:r;r=r,r>r?r@s@r+rr s IRRr.rc8a]tRtRtoRtRt^tRtRtRt Vt R#) _Multinomialia Constrain to nonnegative integer values summing to at most an upper bound. Note due to limitations of the Multinomial distribution, this currently checks the weaker condition ``value.sum(-1) <= upper_bound``. In the future this may be strengthened to ``value.sum(-1) == upper_bound``. TcWnR#rEr)r)rs&&r+rI_Multinomial.__init__$s&r.ctV^8PRR7VPRR7VP8*,#)rrr)rrrrbs&&r+r,_Multinomial.check's1Q|||#quuu}8H8H'HIIr.rN) r3r8r9r:r;r<r=rIr,r>r?r@s@r+r&r&s'KI'JJr.r&c.a]tRtRtoRt^tRtRtVtR#)_LowerTriangulari+z0 Constrain to lower-triangular square matrices. cVP4pW!8HPVPRRR,4PR4^,#)Nrr)trilviewrmin)r)r* value_trils&& r+r,_LowerTriangular.check2sAZZ\ #))%++cr*:U*BCGGKANNr.r7Nr$r@s@r+r,r,+sIOOr.r,c.a]tRtRtoRt^tRtRtVtR#)_LowerCholeskyi7zH Constrain to lower-triangular square matrices with positive diagonals. cVP4pW!8HPVPRRR,4PR4^,pVP RRR7^8PR4^,pW4,#)N)dim1dim2r.rr)r/r0rr1diagonal)r)r*r2rpositive_diagonals&& r+r,_LowerCholesky.check>svZZ\  & &u{{3B'7%'? @ D DR H K #^^"^=AFFrJ1M33r.r7Nr$r@s@r+r5r57sI44r.r5c.a]tRtRtoRt^tRtRtVtR#) _CorrCholeskyiHzq Constrain to lower-triangular square matrices with positive diagonals and each row vector being of unit length. c\P!VP4PVP R4,^ ,p\P P VP4RR7pVR, P4PV4PRR7p\4PV4V,#)r1r?r) r!finfodtypeepssizelinalgnormdetachr"lerr5r,)r)r*tolrow_norm unit_row_norms&& r+r,_CorrCholesky.checkPs KK $ ( (5::b> 9B > <<$$U\\^$<!C,,.11#6::r:B %%e,}<r?rgrhs@@r+rTrTisIIr.rTc6aa]tRtRtoRtV3RltRtVtV;t#)_PositiveSemidefiniteiuz. Constrain to positive-semidefinite matrices. c<\SV`V4pVP4'gV#\PP V4P ^4PR4#r)rHr,rr!rDeigvalshger)r* sym_checkr2s&& r+r,_PositiveSemidefinite.checkzsMGM%( }} ||$$U+..q155b99r.r7r[rhs@@r+r]r]u::r.r]c6aa]tRtRtoRtV3RltRtVtV;t#)_PositiveDefiniteiz* Constrain to positive-definite matrices. c<\SV`V4pVP4'gV#\PP V4P P^4#r)rHr,rr!rD cholesky_exinforras&& r+r,_PositiveDefinite.checksHGM%( }} ||''.3366q99r.r7r[rhs@@r+rfrfrdr.rfcxaa]tRtRtoRtR V3Rllt]V3RlRl4t]V3RlRl4tRt R t Vt V;t #) _Catiz Constraint functor that applies a sequence of constraints `cseq` at the submatrices at dimension `dim`, each of size `lengths[dim]`, in a way compatible with :func:`torch.cat`. c<\;QJdRV4F 'dK RM RM !RV44'g \R4h\V4VnVf^.\ VP4,p\V4Vn\ VP 4\ VP48wd9\R\ VP 4 R\ VP4 R24hW n\SV`!4R#)c3B"TFp\V\4xK R#5irErjr.0cs& r+ _Cat.__init__..;d:a,,dFT1All elements of cseq must be Constraint instancesNz lengths (z) must match cseq (r) rrzlistcseqlenlengthsrrHrI)r)ryrr{r2s&&&&r+rI _Cat.__init__ss;d;sss;d;;; !TU UJ ?cC N*GG} t|| DII . C -..A#dii.AQQRS  r.c <V^8dQhRS[/#rLrO)rQrAs"r+rR_Cat.__annotate__55T5r.c\;QJd&RVP4F 'gK R# R#!RVP44#)c38"TFqPxK R#5irEr<rps& r+rs#_Cat.is_discrete..4)Q==)TFanyryr4s&r+r<_Cat.is_discrete3s4$))4ss4s4s4$))444r.c <V^8dQhRS[/#rLrX)rQrAs"r+rRr~s3333r.c:\RVP44#)c38"TFqPxK R#5irEr=rps& r+rs!_Cat.event_dim..s2 1;; r)maxryr4s&r+r=_Cat.event_dims2 222r.cVP4)VPu;8:dVP48g-M\RVP RVP4 R24h.p^p\VPVP4FIwrEVP VPW54pVP VPV44W5,pKK \P!W P4#zdim z out of range for value with z dimensions) rrzzipryr{narrowappendr,r!r)r)r*checksstartconstrlengthvs&& r+r, _Cat.checks 6599;6 txxj =eiik]+V !$))T\\:NF TXXu5A MM&,,q/ *NE;yy**r.)ryrr{)rN r3r8r9r:r;rIrfr<r=r,r>r?rgrhs@@r+rlrlsA 5533 + +r.rlcxaa]tRtRtoRtR V3Rllt]V3RlRl4t]V3RlRl4tRt R t Vt V;t #) _Stackiz Constraint functor that applies a sequence of constraints `cseq` at the submatrices at dimension `dim`, in a way compatible with :func:`torch.stack`. c<\;QJdRV4F 'dK RM RM !RV44'g \R4h\V4VnW n\ SV`4R#)c3B"TFp\V\4xK R#5irErorps& r+rs"_Stack.__init__..rurvFTrwN)rrzrxryrrHrI)r)ryrr2s&&&r+rI_Stack.__init__sKs;d;sss;d;;; !TU UJ  r.c <V^8dQhRS[/#rLrO)rQrAs"r+rR_Stack.__annotate__rr.c\;QJd&RVP4F 'gK R# R#!RVP44#)c38"TFqPxK R#5irErrps& r+rs%_Stack.is_discrete..rrTFrr4s&r+r<_Stack.is_discreterr.c <V^8dQhRS[/#rLrX)rQrAs"r+rRrs3r.c\RVP44pVPV,^8d V^, pV#)c38"TFqPxK R#5irErrps& r+rs#_Stack.event_dim..s1y!++yr)rryr)r)rs& r+r=_Stack.event_dims41tyy11 88c>A  1HC r.c "VP4)VPu;8:dVP48g-M\RVP RVP4 R24h\VPVP44Uu.uFq!P VPV4NK pp\ P !\W0P4UUu.uFwrEVPV4NK uppVP4#uupiuuppir) rrzrangerCselectr!r"rryr,)r)r*ivsrrs&& r+r, _Stack.checks 6599;6 txxj =eiik]+V 27uzz$((7K1L M1LAll488Q'1L M{{.1"ii.@ A.@V\\!_.@ A488  N As $DD )ryrrrrhs@@r+rrsA 55   r.rgr?)@collections.abcrtypingrr!__all__rrCrrfrmrwrrrrrrrrrrr rrr&r,r5r=rMrTr]rfrlrr r r rrrrrrrrrr r rrr$rrr rrrr!r#rrrr"r7r.r+rs% D ! !  !  !  !   !   ! ! ! ! ! ! ! ! ! !  !! ""#! $%! &'! ()! *+! ,-! ./! 01! 23! 4 5! 67! 89! : ;! < =! >?! @A! H22:,N,N^.4. :. b.pZ.pb.J*.Jb+z+ 2j 2z2z(*(J:$Z$ $ * * Rz RJ:J& Oz O4Z4"=J="  j  I I :J : : :(+:(+V! Z! J L '$ * ))!,&q)# w$"  S!      #s# & *#%!  L -/%' r.