o Me@sJddlZddlmZddlmZddlmZGdddejZddZdS) N) distribution) transform) independentcs2eZdZdZfddZd ddZddZZS) TransformedDistributiona( Applies a sequence of Transforms to a base distribution. Args: base (Distribution): The base distribution. transforms (Sequence[Transform]): A sequence of ``Transform`` . Examples: .. code-block:: python import paddle from paddle.distribution import transformed_distribution d = transformed_distribution.TransformedDistribution( paddle.distribution.Normal(0., 1.), [paddle.distribution.AffineTransform(paddle.to_tensor(1.), paddle.to_tensor(2.))] ) print(d.sample([10])) # Tensor(shape=[10], dtype=float32, place=Place(gpu:0), stop_gradient=True, # [-0.10697651, 3.33609009, -0.86234951, 5.07457638, 0.75925219, # -4.17087793, 2.22579336, -0.93845034, 0.66054249, 1.50957513]) print(d.log_prob(paddle.to_tensor(0.5))) # Tensor(shape=[1], dtype=float32, place=Place(gpu:0), stop_gradient=True, # [-1.64333570]) cs<t|tjstdt|dt|tjs tdt|dtdd|Ds-tdt |}t |j |j |j jkrMtd|j jdt td|j jt |j krdt||j jt |j f}||_||_||j |j }|jjtt |j |j jd }tt||dt |||t ||ddS) Nz1Expected type of 'base' is Distribution, but got .zGExpected type of 'transforms' is Sequence[Transform] or Chain, but got css|] }t|tjVqdS)N) isinstancerZ Transform).0tr \D:\Projects\ConvertPro\env\Lib\site-packages\paddle/distribution/transformed_distribution.py <sz3TransformedDistribution.__init__..z1All element of transforms must be Transform type.z.'base' needs to have shape with size at least z , bug got r)rr Distribution TypeErrortypetypingSequenceallrZChainTransformlenZ batch_shape event_shape_domain event_rank ValueErrorZ base_shaperZ Independent_base _transformsZ forward_shape _codomainmaxsuperr__init__)selfbaseZ transformschainZtransformed_shapeZtransformed_event_rank __class__r r r3s>    z TransformedDistribution.__init__r cCs&|j|}|jD]}||}q |S)zSample from ``TransformedDistribution``. Args: shape (tuple, optional): The sample shape. Defaults to (). Returns: [Tensor]: The sample result. )rsamplerforward)rshapexr r r r r#Rs  zTransformedDistribution.samplecCsd}|}t|j}t|jD]!}||}||jj|jj7}|t| |||jj}|}q|t|j ||t|j j7}|S)zThe log probability evaluated at value. Args: value (Tensor): The value to be evaluated. Returns: Tensor: The log probability. g) rrreversedrZinverserrr_sum_rightmostZforward_log_det_jacobianrlog_prob)rvaluer)yrr r&r r r r)`s$    z TransformedDistribution.log_prob)r )__name__ __module__ __qualname____doc__rr#r) __classcell__r r r!r rs   rcCs"|dkr|tt| dS|S)Nr)sumlistrange)r*nr r r r(xs"r()rZpaddle.distributionrrrr rr(r r r r s     b