3jSSKJrJrJrJrJrJrJrJr SSK r SSK J r SSK J r SSK Jr SSKJrJr SSKJr SSKJr SS KJr \\\4r\\\\\4r\\r"S S \\5rg) )AnyDictIteratorListOptionalTupleUnioncastN)nn) OperationBase)PolicySequential)ImageSequentialBaseTransformMatrixMinIn)InputSequentialOps) ParamItemeye_likec T^\rSrSrSrSS\\S\SS4U4SjjjrS\ SS4S jr SU4S jjr S\\S\\ 4S jr S \S\ 4S jr S\RS\R4SjrSS\RS\\\S\S\\\\4S\\R4 SjjrSS\\\S\4SjjrS\R0S\\4SjrSS\RS\\S\\\\4S\R4SjjrSS\RS\\\S\\\\4S\R4SjjrSrU=r$)PolicyAugmentBase#z Policy-based image augmentation.policytransformation_matrix_modereturnNc|>URU5n[TU]"U6 URU5 [4UlgN)compose_policysuper__init__!_parse_transformation_matrix_moder $_valid_ops_for_transform_computation)selfrrpolicies __class__s W/home/wildlama/miniconda3/lib/python3.13/site-packages/kornia/augmentation/auto/base.pyrPolicyAugmentBase.__init__&s:&&v. (# ../IJFVEX1modulecNURRUR5 gr)_transform_matricesappendtransform_matrix)r!r's r$%_update_transform_matrix_for_valid_op7PolicyAugmentBase._update_transform_matrix_for_valid_op,s   ''(?(?@r&c@>UR5 [TU] 5$)z4Reset cached params and transformation-matrix state.)_reset_transform_matrix_stater clear_state)r!r#s r$r0PolicyAugmentBase.clear_state/s **,w"$$r&cNUVs/sHo RU5PM sn$s snf)z-Compose policy by the provided policy config.)compose_subpolicy_sequential)r!r subpolicys r$r PolicyAugmentBase.compose_policy4s#NTUf11)r?r@) r7get_forward_sequencezipr r get_transformation_matrixrrrdata) r!r9r>r?r@ named_modulesres_mat_r'parammats r$rD+PolicyAugmentBase.get_transformation_matrixGs" >%&GH H9=9R9RSY9Z +/"%m?QVWY"Z KQ*F3F22d8DO#rFrHr's r$rM#PolicyAugmentBase.is_intensity_onlyfsE:>9R9RSY9Z &IA*F3F++--'r& batch_shapecUR5n/nUHBupE[[U5nURU5n[ XF5nUR U5 MD U$)zGenerate per-module parameters for one policy forward pass. Args: batch_shape: Input shape used to sample operation parameters. Returns: Parameters for each selected child module, in execution order. )rBr r forward_parametersrr*)r!rOrFr>namer' mod_paramrIs r$rQ$PolicyAugmentBase.forward_parametersvs\:>9R9R9T "$)LD-0F11+>Id.E MM% *  r&cxUH3nURUR5n[R"XXCS9nM5 U$)zApply a prepared parameter list to the input tensor. Args: input: Input tensor. params: Parameters produced by :meth:`forward_parameters`. extra_args: Optional overrides forwarded to child transforms. Returns: Transformed tensor. r'rIr@) get_submodulerRr transform)r!r9r>r@rIr's r$transform_inputs"PolicyAugmentBase.transform_inputss;E'' 3F&00UjE r&cUR5 Uc%UnURUSS9upVURU5nUHDnURUR5n[ R "XXsS9nURU5 MF X lU$)a#Apply the policy to ``input`` and cache the parameters used. Args: input: Input tensor. params: Optional precomputed parameters. If omitted, parameters are sampled on-the-fly. extra_args: Optional overrides forwarded to child transforms. Returns: Augmented tensor. )) dim_rangerV) r0 autofill_dimrQrWrRrrX"_update_transform_matrix_by_module_params) r!r9r>r@inprH out_shaperIr's r$forwardPolicyAugmentBase.forwards  >C,,SF,CLA,,Y7FE'' 3F&00UjE  3 3F ;   r&)rar )silence)rN)NFNr)NN)__name__ __module__ __qualname____firstlineno____doc__rSUBPOLICY_CONFIGstrrr r,r0rr3torchTensorr<rrboolrrrDrMSizerQrYrd__static_attributes__ __classcell__)r#s@r$rr#s*Yt$45YSVYgkYY Ai(AT ejjT)_*dh\\+/ ?HPQUVY[^V^Q_H` &uy\\+3DO+DYabfgjlogobpYq r&r)typingrrrrrrr r rnr (kornia.augmentation.auto.operations.baser *kornia.augmentation.auto.operations.policyr "kornia.augmentation.container.baserr!kornia.augmentation.container.opsr$kornia.augmentation.container.paramsrkornia.core.opsrfloatintNUMBERrm OP_CONFIGrlrr&r$rsi$KJJ BGX@:$ ucz  #vx// 0  ?T+-ATr&