3jHSSKrSSKJr SSKJr SSKJrJrJrJrJ r "SS\R5r "S S \R5r "S S \R5r SS \S\\S\\S\\\S4S\\RS\\ \\R"44 Sjjrg)N) brownian_base)brownian_interval)OptionalScalarTensorTupleUnioncx^\rSrSrU4SjrS SjrSr\S5r\S5r \S5r \S5r S r U=r $) ReverseBrownianc6>[[U] 5 XlgN)superr __init__ base_brownian)selfr __class__s T/home/wildlama/miniconda3/lib/python3.13/site-packages/torchsde/_brownian/derived.pyrReverseBrownian.__init__s ot-/*c(URU*U*X4S9$N)return_Ureturn_Ar)rtatbrrs r__call__ReverseBrownian.__call__s !!2#sX!QQrcNURRSURS3$)Nz(base_brownian=))r__name__rrs r__repr__ReverseBrownian.__repr__ s'..))*/$:L:L9MQOOrc.URR$r)rdtyper%s rr)ReverseBrownian.dtype#!!'''rc.URR$r)rdevicer%s rr-ReverseBrownian.device's!!(((rc.URR$r)rshaper%s rr0ReverseBrownian.shape+r+rc.URR$r)rlevy_area_approximationr%s rr3'ReverseBrownian.levy_area_approximation/s!!999rrNFF)r$ __module__ __qualname____firstlineno__rr r&propertyr)r-r0r3__static_attributes__ __classcell__rs@rr r sd+R P(())((::rr c^\rSrSrSrSS\S\S\4U4SjjjrSSjr Sr \ S 5r \ S 5r \ S 5r\ S 5rS rU=r$) BrownianPath4zBrownian path, storing every computed value. Useful for speed, when memory isn't a concern. To use: >>> bm = BrownianPath(t0=0.0, w0=torch.zeros(4, 1)) >>> bm(0., 0.5) tensor([[ 0.0733], [-0.5692], [ 0.1872], [-0.3889]]) t0w0 window_sizec >US-nX l[R"XURURUR SS9Ul[[U]'5 g)zwInitialize Brownian path. Arguments: t0: Initial time. w0: Initial state. window_size: Unused; deprecated. rN)r@t1sizer)r- cache_size) _w0rBrownianIntervalr0r)r- _intervalrr>r)rr@rArBrDrs rrBrownianPath.__init__BsN!V*;;rrxx_a_g_gCE99Y]_ lD*,rcdURXX4S9nUcU(dU(dXPR-nU$rrIrGrtrrrouts rr BrownianPath.__call__O-nnQXnI :hx.C rcNURRSURS3$Nz (interval=r#rr$rIr%s rr&BrownianPath.__repr__V%..))**T^^4DAFFrc.URR$rrIr)r%s rr)BrownianPath.dtypeY~~###rc.URR$rrIr-r%s rr-BrownianPath.device]~~$$$rc.URR$rrIr0r%s rr0BrownianPath.shapearZrc.URR$rrIr3r%s rr3$BrownianPath.levy_area_approximatione~~555rrL)r5)r$r6r7r8__doc__rr intrr r&r9r)r-r0r3r:r;r<s@rr>r>4s  -6 -v -C - -G$$%%$$66rr>c^\rSrSrSrSS\S\S\\S\\S\\S\ S \S \S \\ 4U4S jjjr SS jr Sr \ S5r\ S5r\ S5r\ S5rSrU=r$) BrownianTreejaBrownian tree with fixed entropy. Useful when the map from entropy -> Brownian motion shouldn't depend on the locations and order of the query points. (As the usual BrownianInterval does - note that BrownianTree is slower as a result though.) To use: >>> bm = BrownianTree(t0=0.0, w0=torch.zeros(4, 1)) >>> bm(0., 0.5) tensor([[ 0.0733], [-0.5692], [ 0.1872], [-0.3889]], device='cuda:0') r@rArDw1entropytol pool_size cache_depthsafetyc >UcUS-nUcSn OXB- n X l[R"UUURURUR UUUSU S9 Ul[[U]'5 g)a|Initialize the Brownian tree. The random value generation process exploits the parallel random number paradigm and uses `numpy.random.SeedSequence`. The default generator is PCG64 (used by `default_rng`). Arguments: t0: Initial time. w0: Initial state. t1: Terminal time. w1: Terminal state. entropy: Global seed, defaults to `None` for random entropy. tol: Error tolerance before the binary search is terminated; the search depth ~ log2(tol). pool_size: Size of the pooled entropy. This parameter affects the query speed significantly. cache_depth: Unused; deprecated. safety: Unused; deprecated. NrT) r@rDrEr)r-rmrnro halfway_treeW) rGrrHr0r)r-rIrrjr) rr@rArDrlrmrnrorprqrtrs rrBrownianTree.__init__zst4 :aB :AA*;;r?AACBD((CE99DK@CFOIM>? A lD*,rcdURXX4S9nUcU(dU(dXPR-nU$rrLrMs rr BrownianTree.__call__rQrcNURRSURS3$rSrTr%s rr&BrownianTree.__repr__rVrc.URR$rrXr%s rr)BrownianTree.dtyperZrc.URR$rr\r%s rr-BrownianTree.devicer^rc.URR$rr`r%s rr0BrownianTree.shaperZrc.URR$rrcr%s rr3$BrownianTree.levy_area_approximationrerrL)NNNgư> Nr5)r$r6r7r8rgrr rrhfloatrr r&r9r)r-r0r3r:r;r<s@rrjrjjs ")-(,*.""$$%+/+-6+-+-f%+-f%+-#3- +-  +-  +-"+-"%+-+-ZG$$%%$$66rrjyr@rDrE.r)r-c Uc UROUnUc UROUnUc UROUn[R"SXX4US.UD6$)zZReturns a BrownianInterval object with the same size, device, and dtype as a given tensor.)r@rDrEr)r-)r0r)r-rrH)rr@rDrEr)r-kwargss rbrownian_interval_likersPl177D}AGG%EQXXVF  - - l[a lek llr)gg?NNN)torchrrtypesrrr r r BaseBrownianr r>rjrhr)strr-rrrrrs :::m00:<36=--36lU6=--U6r3524=A:>HL mf m'/ m'/ m"*%S/!: m#+5;;"7 m $,E#u||2C,D#E mr