Index _ | A | B | C | D | E | F | G | H | I | K | L | M | N | O | P | R | S | T | U | W _ __add__() (tdhook.hooks.MultiHookHandle method) __call__() (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper method) (tdhook.attribution.lrp_helpers.rules.ParamModifier method) (tdhook.metrics.InfidelityMetric method) (tdhook.metrics.SensitivityMetric method) __enter__() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.contexts.HookingContext method) (tdhook.contexts.HookingContextWithCache method) (tdhook.hooks.MultiHookHandle method) (tdhook.modules.HookedModuleRun method) (tdhook.weights.pruning.PruningContext method) __exit__() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.contexts.HookingContext method) (tdhook.hooks.MultiHookHandle method) (tdhook.modules.HookedModuleRun method) (tdhook.weights.pruning.PruningContext method) __instancecheck__() (tdhook._types.UnraveledMeta method) (tdhook.attribution.lrp_helpers.types.SubclassMeta method) __repr__() (tdhook.latent.CkaEstimator method) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator method) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator method) (tdhook.latent.InformationImbalanceEstimator method) (tdhook.latent.LocalKnnDimensionEstimator method) (tdhook.latent.representation_similarity.cka.CkaEstimator method) (tdhook.latent.representation_similarity.CkaEstimator method) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator method) (tdhook.latent.representation_similarity.InformationImbalanceEstimator method) (tdhook.latent.TwoNnDimensionEstimator method) (tdhook.modules.FunctionModule method) (tdhook.modules.HookedModule method) (tdhook.modules.IntermediateKeysCleaner method) (tdhook.modules.ModuleCall method) (tdhook.modules.ModuleCallWithCache method) (tdhook.modules.PGDModule method) __subclass__ (tdhook.attribution.lrp_helpers.types.Activation attribute) (tdhook.attribution.lrp_helpers.types.AvgPool attribute) (tdhook.attribution.lrp_helpers.types.BatchNorm attribute) (tdhook.attribution.lrp_helpers.types.Convolution attribute) (tdhook.attribution.lrp_helpers.types.ConvolutionStandard attribute) (tdhook.attribution.lrp_helpers.types.ConvolutionTranspose attribute) (tdhook.attribution.lrp_helpers.types.Linear attribute) (tdhook.attribution.lrp_helpers.types.MaxPool attribute) __subclasscheck__() (tdhook.attribution.lrp_helpers.types.SubclassMeta method) _absolute (tdhook.attribution.grad_cam.GradCAM attribute) (tdhook.attribution.GradCAM attribute) (tdhook.attribution.Saliency attribute) (tdhook.attribution.saliency.Saliency attribute) _adapters (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _additional_init_keys (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _additional_keys (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _allow_overwrite (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _alpha (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _alphas (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _amount_to_prune (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _apply_kwargs (tdhook.attribution.lrp_helpers.rules.Rule attribute) _as_distance_input_dtype() (in module tdhook.latent.representation_similarity.information_imbalance) _ascent (tdhook.modules.PGDModule attribute) _attr_key (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _attribution_key (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) _attributor_fn() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) _baseline_key (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) _batch_size (tdhook.latent.probing.estimators.TorchEstimator attribute) _ca_pca() (in module tdhook.latent.dimension_estimation.ca_pca) _cache (tdhook.contexts.HookingContextWithCache attribute) (tdhook.hooks.CacheProxy attribute) (tdhook.modules.HookedModuleRun attribute) _cache_as_output (tdhook.modules.ModuleCallWithCache attribute) _cache_callback (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationAddition attribute) (tdhook.latent.ActivationPatching attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _cache_key (tdhook.modules.ModuleCallWithCache attribute) _cache_ref (tdhook.modules.ModuleCallWithCache attribute) _cached (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _call() (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper method) (tdhook.attribution.lrp_helpers.rules.EpsilonPlus method) _callback (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) _check_callback_signature() (tdhook.hooks.HookFactory static method) _check_hook_signature() (in module tdhook.hooks) _classes_to_hook (tdhook.hooks.MultiHookManager attribute) _classes_to_skip (tdhook.hooks.MultiHookManager attribute) _clean_intermediate_keys (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationAddition attribute) (tdhook.latent.ActivationPatching attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _clear_cache (tdhook.contexts.HookingContextWithCache attribute) _coef (tdhook.latent.MeanDifferenceClassifier attribute) (tdhook.latent.probing.estimators.MeanDifferenceClassifier attribute) (tdhook.latent.probing.MeanDifferenceClassifier attribute) _compute_convergence_delta (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) _compute_convergence_delta_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) _compute_metrics (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _compute_ranks_from_dist() (in module tdhook.latent.representation_similarity.information_imbalance) _compute_steering_vectors() (tdhook.latent.ActivationAddition method) (tdhook.latent.steering_vectors.ActivationAddition method) _contexts (tdhook.contexts.CompositeHookingContextFactory attribute) _create_pair_probe() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) _data (tdhook.modules.HookedModuleRun attribute) _data_preprocess_callback (tdhook.latent.Probe attribute) (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.Probe attribute) (tdhook.latent.probing.ProbeManager attribute) _decision_function() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.MeanDifferenceClassifier method) _default_data_preprocess_callback() (tdhook.latent.Probe method) (tdhook.latent.probing.managers.Probe method) (tdhook.latent.probing.Probe method) _device (tdhook.latent.probing.estimators.TorchEstimator attribute) _dim_from_ca_pca() (in module tdhook.latent.dimension_estimation.ca_pca) _dim_from_eigenvalues_maxgap() (in module tdhook.latent.dimension_estimation.local_pca) _dim_from_eigenvalues_ratio() (in module tdhook.latent.dimension_estimation.local_pca) _directional_imbalance() (in module tdhook.latent.representation_similarity.information_imbalance) _directions (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _ensure_in_context() (tdhook.modules.HookedModuleRun method) _ensure_sklearn() (in module tdhook._optional_deps) _enter() (tdhook.contexts.HookingContext method) (tdhook.contexts.HookingContextWithCache method) (tdhook.weights.pruning.PruningContext method) _epochs (tdhook.latent.probing.estimators.TorchEstimator attribute) _estimator (tdhook.latent.Probe attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.Probe attribute) _estimator_class (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _estimator_kwargs (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _estimators (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _fit_callback (tdhook.latent.Probe attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.Probe attribute) _fit_metrics (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _flatten (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _grad_attr() (tdhook.attribution.grad_cam.GradCAM method) (tdhook.attribution.GradCAM method) (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation method) (tdhook.attribution.GuidedBackpropagation method) (tdhook.attribution.integrated_gradients.IntegratedGradients method) (tdhook.attribution.IntegratedGradients method) (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.attribution.Saliency method) (tdhook.attribution.saliency.Saliency method) _grad_enabled (tdhook.modules.HookedModuleRun attribute) _grad_key (tdhook.modules.PGDModule attribute) _h1_key (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _h2_key (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _handle (tdhook.contexts.HookingContext attribute) _handles (tdhook.hooks.MultiHookHandle attribute) (tdhook.modules.HookedModuleRun attribute) _has_torchvision (in module tdhook.attribution.lrp_helpers.types) _hook (tdhook.contexts.HookingContext attribute) _hook_manager (tdhook.attribution.guided_backpropagation.GuidedBackpropagation attribute) (tdhook.attribution.GuidedBackpropagation attribute) (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _hook_module() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation method) (tdhook.attribution.GuidedBackpropagation method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) (tdhook.latent.activation_caching.ActivationCaching method) (tdhook.latent.activation_patching.ActivationPatching method) (tdhook.latent.ActivationAddition method) (tdhook.latent.ActivationCaching method) (tdhook.latent.ActivationPatching method) (tdhook.latent.Probing method) (tdhook.latent.probing.context.Probing method) (tdhook.latent.probing.Probing method) (tdhook.latent.steering_vectors.ActivationAddition method) (tdhook.latent.steering_vectors.SteeringVectors method) (tdhook.latent.SteeringVectors method) (tdhook.weights.Adapters method) (tdhook.weights.adapters.Adapters method) _hooked_module (tdhook.contexts.HookingContext attribute) _hooked_module_class (tdhook.contexts.HookingContextFactory attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) (tdhook.weights.task_vectors.TaskVectors attribute) (tdhook.weights.TaskVectors attribute) _hooked_module_kwargs (tdhook.contexts.HookingContextFactory attribute) _hooking_context (tdhook.modules.HookedModule attribute) _hooking_context_class (tdhook.contexts.HookingContextFactory attribute) (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) (tdhook.weights.task_vectors.TaskVectors attribute) (tdhook.weights.TaskVectors attribute) _hooking_context_kwargs (tdhook.contexts.HookingContextFactory attribute) _importance_callback (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _in_context (tdhook.contexts.HookingContext attribute) (tdhook.modules.HookedModuleRun attribute) _in_key (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _in_keys (tdhook.contexts.HookingContext attribute) _information_imbalance() (in module tdhook.latent.representation_similarity.information_imbalance) _init_attr_cache_in (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_grads (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_inputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _init_attr_targets (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _input_modules (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _intercept (tdhook.latent.MeanDifferenceClassifier attribute) (tdhook.latent.probing.estimators.MeanDifferenceClassifier attribute) (tdhook.latent.probing.MeanDifferenceClassifier attribute) _intermediate_keys (tdhook.modules.IntermediateKeysCleaner attribute) _key (tdhook.hooks.CacheProxy attribute) (tdhook.hooks.EarlyStoppingException attribute) _key_pattern (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _key_to_probes (tdhook.latent.BilinearProbeManager attribute) (tdhook.latent.probing.BilinearProbeManager attribute) (tdhook.latent.probing.managers.BilinearProbeManager attribute) _linear_cka() (in module tdhook.latent.representation_similarity.cka) _local_knn() (in module tdhook.latent.dimension_estimation.local_knn) _local_pca() (in module tdhook.latent.dimension_estimation.local_pca) _loss_fn() (tdhook.latent.probing.estimators.TorchEstimator method) _lr (tdhook.latent.probing.estimators.TorchEstimator attribute) _managed_by_context_manager (tdhook.contexts.HookingContext attribute) _max_value (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _method (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _min_value (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.modules.PGDModule attribute) _modify_fn (tdhook.attribution.lrp_helpers.rules.ParamModifier attribute) _module (tdhook.modules.HookedModuleRun attribute) _module_ref (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle attribute) _modules_to_attribute (tdhook.attribution.grad_cam.GradCAM attribute) (tdhook.attribution.GradCAM attribute) _modules_to_maximise (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) _modules_to_patch (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _modules_to_prune (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _modules_to_steer (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) (tdhook.latent.steering_vectors.SteeringVectors attribute) (tdhook.latent.SteeringVectors attribute) _multiply_by_inputs (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline attribute) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline attribute) (tdhook.attribution.guided_backpropagation.GuidedBackpropagation attribute) (tdhook.attribution.GuidedBackpropagation attribute) (tdhook.attribution.Saliency attribute) (tdhook.attribution.saliency.Saliency attribute) _multiply_by_inputs_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) _n_steps (tdhook.attribution.activation_maximisation.ActivationMaximisation attribute) (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) (tdhook.modules.PGDModule attribute) _name (tdhook.modules.HookedModuleRun attribute) _negative_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _normalize (tdhook.latent.MeanDifferenceClassifier attribute) (tdhook.latent.probing.estimators.MeanDifferenceClassifier attribute) (tdhook.latent.probing.MeanDifferenceClassifier attribute) _num_classes (tdhook.latent.probing.estimators.TorchEstimator attribute) _old_weights (tdhook.weights.pruning.PruningContext attribute) _out_key (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) _out_keys (tdhook.contexts.HookingContext attribute) _outer_cache (tdhook.modules.HookedModuleRun attribute) _output_grad_callbacks (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _pair_probes (tdhook.latent.BilinearProbeManager attribute) (tdhook.latent.probing.BilinearProbeManager attribute) (tdhook.latent.probing.managers.BilinearProbeManager attribute) _pairs (tdhook.latent.BilinearProbeManager attribute) (tdhook.latent.probing.BilinearProbeManager attribute) (tdhook.latent.probing.managers.BilinearProbeManager attribute) _patch_fn (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _patch_key (tdhook.latent.activation_patching.ActivationPatching attribute) (tdhook.latent.ActivationPatching attribute) _pattern (tdhook.hooks.MultiHookManager attribute) _perturb_data() (tdhook.metrics.InfidelityMetric method) (tdhook.metrics.SensitivityMetric method) _pgd_step() (tdhook.modules.PGDModule method) _positive_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _pre_factories (tdhook.contexts.HookingContext attribute) _predict_callback (tdhook.latent.Probe attribute) (tdhook.latent.probing.managers.Probe attribute) (tdhook.latent.probing.Probe attribute) _predict_metrics (tdhook.latent.ProbeManager attribute) (tdhook.latent.probing.managers.ProbeManager attribute) (tdhook.latent.probing.ProbeManager attribute) _prepare (tdhook.contexts.HookingContext attribute) _prepare_module() (tdhook.attribution.activation_maximisation.ActivationMaximisation method) (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) (tdhook.latent.activation_patching.ActivationPatching method) (tdhook.latent.ActivationAddition method) (tdhook.latent.ActivationPatching method) (tdhook.latent.steering_vectors.ActivationAddition method) (tdhook.weights.Pruning method) (tdhook.weights.pruning.Pruning method) _probe_factory (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) _reduce_baselines_fn() (tdhook.attribution.gradient_helpers.common.GradientAttributionWithBaseline method) (tdhook.attribution.gradient_helpers.GradientAttributionWithBaseline method) (tdhook.attribution.integrated_gradients.IntegratedGradients method) (tdhook.attribution.IntegratedGradients method) _ref (tdhook.hooks.MutableWeakRef attribute) _reg_exp (tdhook.hooks.MultiHookManager attribute) _register_inputs_fn() (tdhook.attribution.gradient_helpers.common.GradientAttribution method) (tdhook.attribution.gradient_helpers.GradientAttribution method) _relative (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) (tdhook.weights.Adapters attribute) (tdhook.weights.adapters.Adapters attribute) _relative_path (tdhook.modules.HookedModule attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _resolve_k() (in module tdhook.latent.dimension_estimation.local_knn) _restore (tdhook.contexts.HookingContext attribute) _restore_module() (tdhook.attribution.LRP method) (tdhook.attribution.lrp.LRP method) (tdhook.contexts.CompositeHookingContextFactory method) (tdhook.contexts.HookingContextFactory method) _rule (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle attribute) _rule_mapper (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _rules (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _run() (tdhook.latent.BilinearProbe method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.managers.BilinearProbe method) _run_callback (tdhook.modules.HookedModuleRun attribute) _save_cache (tdhook.modules.HookedModuleRun attribute) _select_fn (tdhook.attribution.lrp_helpers.rules.ParamModifier attribute) _sep (tdhook.modules.HookedModuleRun attribute) _skip_modules (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) (tdhook.weights.Pruning attribute) (tdhook.weights.pruning.Pruning attribute) _spawn (tdhook.contexts.HookingContext attribute) _spawn_hooked_module() (tdhook.contexts.HookingContextFactory method) _stabilizer (tdhook.attribution.lrp_helpers.rules.BaseRuleMapper attribute) _stack (tdhook.contexts.HookingContext attribute) _steer_fn (tdhook.latent.steering_vectors.SteeringVectors attribute) (tdhook.latent.SteeringVectors attribute) _steer_key (tdhook.latent.ActivationAddition attribute) (tdhook.latent.steering_vectors.ActivationAddition attribute) _step_sizes (tdhook.attribution.integrated_gradients.IntegratedGradients attribute) (tdhook.attribution.IntegratedGradients attribute) _target_modules (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _td (tdhook.hooks.TensorDictRef attribute) _td_fn (tdhook.modules.FunctionModule attribute) _td_module (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) (tdhook.modules.PGDModule attribute) _twonn() (in module tdhook.latent.dimension_estimation.twonn) _use_inputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _use_nested_keys (tdhook.latent.activation_caching.ActivationCaching attribute) (tdhook.latent.ActivationCaching attribute) _use_outputs (tdhook.attribution.gradient_helpers.common.GradientAttribution attribute) (tdhook.attribution.gradient_helpers.GradientAttribution attribute) _use_sign (tdhook.modules.PGDModule attribute) _validate_inputs() (in module tdhook.latent.representation_similarity.cka) (in module tdhook.latent.representation_similarity.information_imbalance) _verbose (tdhook.latent.probing.estimators.TorchEstimator attribute) _waiting_active (tdhook.latent.BilinearProbe attribute) (tdhook.latent.probing.BilinearProbe attribute) (tdhook.latent.probing.managers.BilinearProbe attribute) _warn_on_missing_rule (tdhook.attribution.LRP attribute) (tdhook.attribution.lrp.LRP attribute) _weights (tdhook.weights.task_vectors.TaskVectorsModule attribute) _working_key (tdhook.modules.PGDModule attribute) A AbstractFunctionMeta (class in tdhook.attribution.lrp_helpers.rules) Activation (class in tdhook.attribution.lrp_helpers.types) ActivationAddition (class in tdhook.latent) (class in tdhook.latent.steering_vectors) ActivationCaching (class in tdhook.latent) (class in tdhook.latent.activation_caching) ActivationMaximisation (class in tdhook.attribution.activation_maximisation) ActivationPatching (class in tdhook.latent) (class in tdhook.latent.activation_patching) Adapters (class in tdhook.weights) (class in tdhook.weights.adapters) adapters (tdhook.weights.adapters.HookedModuleWithAdapters attribute) after_all() (tdhook.latent.BilinearProbe method) (tdhook.latent.BilinearProbeManager method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbe method) (tdhook.latent.probing.managers.BilinearProbeManager method) AHQKVRule (class in tdhook.attribution.lrp_helpers.rules) alpha (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) AlphaBetaRule (class in tdhook.attribution.lrp_helpers.rules) alphas (tdhook.weights.task_vectors.TaskVectorsContext attribute) approximation_parameters() (in module tdhook.attribution.gradient_helpers.helpers) AvgPool (class in tdhook.attribution.lrp_helpers.types) B backward() (tdhook.attribution.lrp_helpers.rules.AHQKVRule static method) (tdhook.attribution.lrp_helpers.rules.AlphaBetaRule static method) (tdhook.attribution.lrp_helpers.rules.EpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.IgnoreRule static method) (tdhook.attribution.lrp_helpers.rules.LayerNormRule static method) (tdhook.attribution.lrp_helpers.rules.PassRule static method) (tdhook.attribution.lrp_helpers.rules.PseudoIdentityRule static method) (tdhook.attribution.lrp_helpers.rules.Rule static method) (tdhook.attribution.lrp_helpers.rules.SoftmaxEpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.StopRule static method) (tdhook.attribution.lrp_helpers.rules.UniformEpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.UniformRule static method) (tdhook.attribution.lrp_helpers.rules.WSquareRule static method) BaseRuleMapper (class in tdhook.attribution.lrp_helpers.rules) BatchNorm (class in tdhook.attribution.lrp_helpers.types) before_all() (tdhook.latent.BilinearProbe method) (tdhook.latent.BilinearProbeManager method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbe method) (tdhook.latent.probing.managers.BilinearProbeManager method) bilinear (tdhook.latent.probing.estimators.BilinearEstimator attribute) BilinearEstimator (class in tdhook.latent.probing.estimators) BilinearProbe (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.managers) BilinearProbeManager (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.managers) C cache (tdhook.contexts.HookingContextWithCache property) (tdhook.modules.HookedModuleRun property) cache_ref (tdhook.modules.ModuleCallWithCache property) CacheProxy (class in tdhook.hooks) CaPcaDimensionEstimator (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.ca_pca) CkaEstimator (class in tdhook.latent) (class in tdhook.latent.representation_similarity) (class in tdhook.latent.representation_similarity.cka) clear() (tdhook.contexts.HookingContextWithCache method) coef_ (tdhook.latent.MeanDifferenceClassifier property) (tdhook.latent.probing.estimators.MeanDifferenceClassifier property) (tdhook.latent.probing.MeanDifferenceClassifier property) CompositeHookingContextFactory (class in tdhook.contexts) compute_alpha() (tdhook.weights.task_vectors.TaskVectorsContext method) Convolution (class in tdhook.attribution.lrp_helpers.types) ConvolutionStandard (class in tdhook.attribution.lrp_helpers.types) ConvolutionTranspose (class in tdhook.attribution.lrp_helpers.types) criterion (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) D default_classes_to_hook (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) default_classes_to_skip (tdhook.latent.Probing attribute) (tdhook.latent.probing.context.Probing attribute) (tdhook.latent.probing.Probing attribute) default_skip() (tdhook.attribution.LRP static method) (tdhook.attribution.lrp.LRP static method) (tdhook.weights.Pruning static method) (tdhook.weights.pruning.Pruning static method) dim (tdhook.attribution.lrp_helpers.layers.Sum attribute) DimsConfig (class in tdhook.attribution.grad_cam) DIRECTION_TO_PARAMS (in module tdhook.hooks) DIRECTION_TO_RETURN (in module tdhook.hooks) DIRECTION_TO_RETURN_INDEX (in module tdhook.hooks) DIRECTION_TO_TYPE (in module tdhook.hooks) disable() (tdhook.contexts.HookingContext method) disable_context() (tdhook.modules.HookedModule method) disable_context_hooks() (tdhook.modules.HookedModule method) disable_hooks() (tdhook.contexts.HookingContext method) E EarlyStoppingException eps (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) epsilon (tdhook.attribution.lrp_helpers.rules.EpsilonRule property) EpsilonPlus (class in tdhook.attribution.lrp_helpers.rules) EpsilonRule (class in tdhook.attribution.lrp_helpers.rules) estimators (tdhook.latent.ProbeManager property) (tdhook.latent.probing.managers.ProbeManager property) (tdhook.latent.probing.ProbeManager property) F feature_sum_dims (tdhook.attribution.grad_cam.DimsConfig attribute) fit() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.TorchEstimator method) (tdhook.latent.probing.MeanDifferenceClassifier method) fit_metrics (tdhook.latent.ProbeManager property) (tdhook.latent.probing.managers.ProbeManager property) (tdhook.latent.probing.ProbeManager property) FlatRule (class in tdhook.attribution.lrp_helpers.rules) flatten_select_reshape_call() (in module tdhook.modules) forward() (tdhook.attribution.lrp_helpers.layers.Sum method) (tdhook.attribution.lrp_helpers.rules.AHQKVRule static method) (tdhook.attribution.lrp_helpers.rules.AlphaBetaRule static method) (tdhook.attribution.lrp_helpers.rules.EpsilonRule static method) (tdhook.attribution.lrp_helpers.rules.IgnoreRule static method) (tdhook.attribution.lrp_helpers.rules.LayerNormRule static method) (tdhook.attribution.lrp_helpers.rules.PassRule static method) (tdhook.attribution.lrp_helpers.rules.PseudoIdentityRule static method) (tdhook.attribution.lrp_helpers.rules.Rule static method) (tdhook.attribution.lrp_helpers.rules.StopRule static method) (tdhook.attribution.lrp_helpers.rules.UniformRule static method) (tdhook.attribution.lrp_helpers.rules.WSquareRule static method) (tdhook.latent.CkaEstimator method) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator method) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator method) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator method) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator method) (tdhook.latent.InformationImbalanceEstimator method) (tdhook.latent.LinearEstimator method) (tdhook.latent.LocalKnnDimensionEstimator method) (tdhook.latent.LowRankBilinearEstimator method) (tdhook.latent.probing.estimators.BilinearEstimator method) (tdhook.latent.probing.estimators.LinearEstimator method) (tdhook.latent.probing.estimators.LowRankBilinearEstimator method) (tdhook.latent.probing.LinearEstimator method) (tdhook.latent.probing.LowRankBilinearEstimator method) (tdhook.latent.representation_similarity.cka.CkaEstimator method) (tdhook.latent.representation_similarity.CkaEstimator method) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator method) (tdhook.latent.representation_similarity.InformationImbalanceEstimator method) (tdhook.latent.TwoNnDimensionEstimator method) (tdhook.modules.FunctionModule method) (tdhook.modules.HookedModule method) (tdhook.modules.IntermediateKeysCleaner method) (tdhook.modules.ModuleCall method) (tdhook.modules.ModuleCallWithCache method) (tdhook.modules.PGDModule method) from_modifiers() (tdhook.attribution.lrp_helpers.rules.ParamModifier static method) from_module() (tdhook.modules.HookedModule class method) FunctionModule (class in tdhook.modules) G gauss_legendre_builders() (in module tdhook.attribution.gradient_helpers.helpers) get() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_best_device() (in module tdhook.modules) get_control_adequacy (tdhook.weights.task_vectors.TaskVectorsContext attribute) get_forget_vector() (tdhook.weights.task_vectors.TaskVectorsModule method) get_grad() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_grad_output() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_input() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) get_task_vector() (tdhook.weights.task_vectors.TaskVectorsModule method) get_test_accuracy (tdhook.weights.task_vectors.TaskVectorsContext attribute) get_weights() (tdhook.weights.task_vectors.TaskVectorsModule method) GradCAM (class in tdhook.attribution) (class in tdhook.attribution.grad_cam) GradientAttribution (class in tdhook.attribution.gradient_helpers) (class in tdhook.attribution.gradient_helpers.common) GradientAttributionWithBaseline (class in tdhook.attribution.gradient_helpers) (class in tdhook.attribution.gradient_helpers.common) GuidedBackpropagation (class in tdhook.attribution) (class in tdhook.attribution.guided_backpropagation) H HookDirection (in module tdhook.hooks) HookedModule (class in tdhook.modules) HookedModuleRun (class in tdhook.modules) HookedModuleWithAdapters (class in tdhook.weights.adapters) HookFactory (class in tdhook.hooks) hooking_context (tdhook.modules.HookedModule property) HookingContext (class in tdhook.contexts) HookingContextFactory (class in tdhook.contexts) HookingContextWithCache (class in tdhook.contexts) I IgnoreRule (class in tdhook.attribution.lrp_helpers.rules) in_key (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) in_key_a (tdhook.latent.CkaEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) in_key_b (tdhook.latent.CkaEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) in_keys (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) (tdhook.modules.FunctionModule attribute) (tdhook.modules.IntermediateKeysCleaner attribute) (tdhook.modules.ModuleCall attribute) (tdhook.modules.ModuleCallWithCache attribute) (tdhook.modules.PGDModule attribute) InfidelityMetric (class in tdhook.metrics) InformationImbalanceEstimator (class in tdhook.latent) (class in tdhook.latent.representation_similarity) (class in tdhook.latent.representation_similarity.information_imbalance) init_attr_targets_with_labels() (tdhook.attribution.integrated_gradients.IntegratedGradients static method) (tdhook.attribution.IntegratedGradients static method) IntegratedGradients (class in tdhook.attribution) (class in tdhook.attribution.integrated_gradients) intercept_ (tdhook.latent.MeanDifferenceClassifier property) (tdhook.latent.probing.estimators.MeanDifferenceClassifier property) (tdhook.latent.probing.MeanDifferenceClassifier property) IntermediateKeysCleaner (class in tdhook.modules) is_waiting (tdhook.latent.BilinearProbe property) (tdhook.latent.probing.BilinearProbe property) (tdhook.latent.probing.managers.BilinearProbe property) K k (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) kernel (tdhook.latent.CkaEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) key_pattern (tdhook.latent.activation_caching.ActivationCaching property) (tdhook.latent.ActivationCaching property) (tdhook.latent.BilinearProbeManager property) (tdhook.latent.Probing property) (tdhook.latent.probing.BilinearProbeManager property) (tdhook.latent.probing.context.Probing property) (tdhook.latent.probing.managers.BilinearProbeManager property) (tdhook.latent.probing.Probing property) L LayerNormRule (class in tdhook.attribution.lrp_helpers.rules) left (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) Linear (class in tdhook.attribution.lrp_helpers.types) linear (tdhook.latent.LinearEstimator attribute) (tdhook.latent.probing.estimators.LinearEstimator attribute) (tdhook.latent.probing.LinearEstimator attribute) linear1 (tdhook.latent.LowRankBilinearEstimator attribute) (tdhook.latent.probing.estimators.LowRankBilinearEstimator attribute) (tdhook.latent.probing.LowRankBilinearEstimator attribute) linear2 (tdhook.latent.LowRankBilinearEstimator attribute) (tdhook.latent.probing.estimators.LowRankBilinearEstimator attribute) (tdhook.latent.probing.LowRankBilinearEstimator attribute) LinearEstimator (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.estimators) LocalKnnDimensionEstimator (class in tdhook.latent) (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.local_knn) LocalPcaDimensionEstimator (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.local_pca) LowRankBilinearEstimator (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.estimators) LRP (class in tdhook.attribution) (class in tdhook.attribution.lrp) M make_caching_hook() (tdhook.hooks.HookFactory static method) make_reading_hook() (tdhook.hooks.HookFactory static method) make_setting_hook() (tdhook.hooks.HookFactory static method) make_stopping_hook() (tdhook.hooks.HookFactory static method) MaxPool (class in tdhook.attribution.lrp_helpers.types) MeanDifferenceClassifier (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.estimators) merge_paths() (in module tdhook.hooks) middle (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) module tdhook tdhook._optional_deps tdhook._types tdhook.attribution tdhook.attribution.activation_maximisation tdhook.attribution.grad_cam tdhook.attribution.gradient_helpers tdhook.attribution.gradient_helpers.common tdhook.attribution.gradient_helpers.helpers tdhook.attribution.guided_backpropagation tdhook.attribution.integrated_gradients tdhook.attribution.lrp tdhook.attribution.lrp_helpers tdhook.attribution.lrp_helpers.layers tdhook.attribution.lrp_helpers.rules tdhook.attribution.lrp_helpers.types tdhook.attribution.saliency tdhook.auto tdhook.contexts tdhook.hooks tdhook.latent tdhook.latent.activation_caching tdhook.latent.activation_patching tdhook.latent.dimension_estimation tdhook.latent.dimension_estimation._utils tdhook.latent.dimension_estimation.ca_pca tdhook.latent.dimension_estimation.local_knn tdhook.latent.dimension_estimation.local_pca tdhook.latent.dimension_estimation.twonn tdhook.latent.probing tdhook.latent.probing.context tdhook.latent.probing.estimators tdhook.latent.probing.managers tdhook.latent.representation_similarity tdhook.latent.representation_similarity.cka tdhook.latent.representation_similarity.information_imbalance tdhook.latent.steering_vectors tdhook.metrics tdhook.modules tdhook.sources tdhook.weights tdhook.weights.adapters tdhook.weights.pruning tdhook.weights.task_vectors ModuleCall (class in tdhook.modules) ModuleCallWithCache (class in tdhook.modules) MultiHookHandle (class in tdhook.hooks) MultiHookManager (class in tdhook.hooks) MutableWeakRef (class in tdhook.hooks) N n_perturb_samples (tdhook.metrics.InfidelityMetric attribute) O out_key (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) out_key_a_to_b (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) out_key_b_to_a (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) out_keys (tdhook.latent.CkaEstimator attribute) (tdhook.latent.dimension_estimation.ca_pca.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.CaPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_knn.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.local_pca.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalKnnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.LocalPcaDimensionEstimator attribute) (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.LocalKnnDimensionEstimator attribute) (tdhook.latent.representation_similarity.cka.CkaEstimator attribute) (tdhook.latent.representation_similarity.CkaEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) (tdhook.modules.FunctionModule attribute) (tdhook.modules.IntermediateKeysCleaner attribute) (tdhook.modules.ModuleCall attribute) (tdhook.modules.PGDModule attribute) P p (tdhook.latent.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.information_imbalance.InformationImbalanceEstimator attribute) (tdhook.latent.representation_similarity.InformationImbalanceEstimator attribute) ParamModifier (class in tdhook.attribution.lrp_helpers.rules) PassRule (class in tdhook.attribution.lrp_helpers.rules) pattern (tdhook.hooks.MultiHookManager property) perturb_radius (tdhook.metrics.SensitivityMetric attribute) PGDModule (class in tdhook.modules) predict() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.TorchEstimator method) (tdhook.latent.probing.MeanDifferenceClassifier method) predict_metrics (tdhook.latent.ProbeManager property) (tdhook.latent.probing.managers.ProbeManager property) (tdhook.latent.probing.ProbeManager property) predict_proba() (tdhook.latent.MeanDifferenceClassifier method) (tdhook.latent.probing.estimators.MeanDifferenceClassifier method) (tdhook.latent.probing.MeanDifferenceClassifier method) prepare() (tdhook.contexts.HookingContextFactory method) Probe (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.context) (class in tdhook.latent.probing.managers) probe_factory() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.ProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) (tdhook.latent.probing.managers.ProbeManager method) (tdhook.latent.probing.ProbeManager method) ProbeManager (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.managers) Probing (class in tdhook.latent) (class in tdhook.latent.probing) (class in tdhook.latent.probing.context) Pruning (class in tdhook.weights) (class in tdhook.weights.pruning) PruningContext (class in tdhook.weights.pruning) PseudoIdentityRule (class in tdhook.attribution.lrp_helpers.rules) R raise_for_unconserved_rel_factory() (in module tdhook.attribution.lrp_helpers.rules) register() (tdhook.attribution.lrp_helpers.rules.IgnoreRule method) (tdhook.attribution.lrp_helpers.rules.Rule method) register_hook() (tdhook.hooks.MultiHookManager method) register_hook_to_module() (in module tdhook.hooks) register_submodule_hook() (tdhook.modules.HookedModule method) relative_path (tdhook.modules.HookedModule property) RemovableHandleProtocol (class in tdhook.hooks) RemovableRuleHandle (class in tdhook.attribution.lrp_helpers.rules) remove() (tdhook.attribution.lrp_helpers.rules.RemovableRuleHandle method) (tdhook.hooks.MultiHookHandle method) (tdhook.hooks.RemovableHandleProtocol method) requires_sklearn() (in module tdhook._optional_deps) reset_estimators() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.ProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) (tdhook.latent.probing.managers.ProbeManager method) (tdhook.latent.probing.ProbeManager method) reset_metrics() (tdhook.latent.BilinearProbeManager method) (tdhook.latent.ProbeManager method) (tdhook.latent.probing.BilinearProbeManager method) (tdhook.latent.probing.managers.BilinearProbeManager method) (tdhook.latent.probing.managers.ProbeManager method) (tdhook.latent.probing.ProbeManager method) resolve() (tdhook.hooks.CacheProxy method) (tdhook.hooks.MutableWeakRef method) (tdhook.hooks.TensorDictRef method) resolve_submodule_path() (in module tdhook.hooks) restore() (tdhook.modules.HookedModule method) return_xy (tdhook.latent.dimension_estimation.twonn.TwoNnDimensionEstimator attribute) (tdhook.latent.dimension_estimation.TwoNnDimensionEstimator attribute) (tdhook.latent.TwoNnDimensionEstimator attribute) Riemann (class in tdhook.attribution.gradient_helpers.helpers) riemann_builders() (in module tdhook.attribution.gradient_helpers.helpers) right (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) Rule (class in tdhook.attribution.lrp_helpers.rules) run() (tdhook.modules.HookedModule method) S Saliency (class in tdhook.attribution) (class in tdhook.attribution.saliency) save() (tdhook.modules.HookedModuleRun method) save_grad() (tdhook.modules.HookedModuleRun method) save_grad_output() (tdhook.modules.HookedModuleRun method) save_input() (tdhook.modules.HookedModuleRun method) select_all() (tdhook.attribution.lrp_helpers.rules.ParamModifier static method) SensitivityMetric (class in tdhook.metrics) set() (tdhook.hooks.MutableWeakRef method) (tdhook.hooks.TensorDictRef method) (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_grad() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_grad_output() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) set_input() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) SoftmaxEpsilonRule (class in tdhook.attribution.lrp_helpers.rules) sorted_neighbors() (in module tdhook.latent.dimension_estimation._utils) stabilize() (in module tdhook.attribution.lrp_helpers.rules) state_dicts() (tdhook.attribution.lrp_helpers.rules.ParamModifier method) SteeringVectors (class in tdhook.latent) (class in tdhook.latent.steering_vectors) step() (tdhook.latent.BilinearProbe method) (tdhook.latent.Probe method) (tdhook.latent.probing.BilinearProbe method) (tdhook.latent.probing.context.Probe method) (tdhook.latent.probing.managers.BilinearProbe method) (tdhook.latent.probing.managers.Probe method) (tdhook.latent.probing.Probe method) stop() (tdhook.modules.HookedModule method) (tdhook.modules.HookedModuleRun method) StopRule (class in tdhook.attribution.lrp_helpers.rules) SubclassMeta (class in tdhook.attribution.lrp_helpers.types) submodule_path_to_name() (in module tdhook.hooks) Sum (class in tdhook.attribution.lrp_helpers.layers) SUPPORTED_METHODS (in module tdhook.attribution.gradient_helpers.helpers) SUPPORTED_RIEMANN_METHODS (in module tdhook.attribution.gradient_helpers.helpers) T T (in module tdhook.hooks) TaskVectors (class in tdhook.weights) (class in tdhook.weights.task_vectors) TaskVectorsContext (class in tdhook.weights.task_vectors) TaskVectorsModule (class in tdhook.weights.task_vectors) tdhook module tdhook._optional_deps module tdhook._types module tdhook.attribution module tdhook.attribution.activation_maximisation module tdhook.attribution.grad_cam module tdhook.attribution.gradient_helpers module tdhook.attribution.gradient_helpers.common module tdhook.attribution.gradient_helpers.helpers module tdhook.attribution.guided_backpropagation module tdhook.attribution.integrated_gradients module tdhook.attribution.lrp module tdhook.attribution.lrp_helpers module tdhook.attribution.lrp_helpers.layers module tdhook.attribution.lrp_helpers.rules module tdhook.attribution.lrp_helpers.types module tdhook.attribution.saliency module tdhook.auto module tdhook.contexts module tdhook.hooks module tdhook.latent module tdhook.latent.activation_caching module tdhook.latent.activation_patching module tdhook.latent.dimension_estimation module tdhook.latent.dimension_estimation._utils module tdhook.latent.dimension_estimation.ca_pca module tdhook.latent.dimension_estimation.local_knn module tdhook.latent.dimension_estimation.local_pca module tdhook.latent.dimension_estimation.twonn module tdhook.latent.probing module tdhook.latent.probing.context module tdhook.latent.probing.estimators module tdhook.latent.probing.managers module tdhook.latent.representation_similarity module tdhook.latent.representation_similarity.cka module tdhook.latent.representation_similarity.information_imbalance module tdhook.latent.steering_vectors module tdhook.metrics module tdhook.modules module tdhook.sources module tdhook.weights module tdhook.weights.adapters module tdhook.weights.pruning module tdhook.weights.task_vectors module TensorDictRef (class in tdhook.hooks) TorchEstimator (class in tdhook.latent.probing.estimators) torchvision (in module tdhook.attribution.lrp_helpers.types) trapezoid (tdhook.attribution.gradient_helpers.helpers.Riemann attribute) TwoNnDimensionEstimator (class in tdhook.latent) (class in tdhook.latent.dimension_estimation) (class in tdhook.latent.dimension_estimation.twonn) U UniformEpsilonRule (class in tdhook.attribution.lrp_helpers.rules) UniformRule (class in tdhook.attribution.lrp_helpers.rules) UnraveledKey (class in tdhook._types) UnraveledMeta (class in tdhook._types) unregister() (tdhook.attribution.lrp_helpers.rules.IgnoreRule method) (tdhook.attribution.lrp_helpers.rules.Rule method) W weight_pooling_dims (tdhook.attribution.grad_cam.DimsConfig attribute) with_applied_vectors() (tdhook.weights.task_vectors.TaskVectorsModule method) WSquareRule (class in tdhook.attribution.lrp_helpers.rules)
