Struct ActivationVisualizer 

pub struct ActivationVisualizer { /* private fields */ }

Activation visualizer for inspecting layer outputs

Implementations

impl ActivationVisualizer

pub fn new() -> Self

Create a new activation visualizer

Example

use trustformers_debug::ActivationVisualizer;

let visualizer = ActivationVisualizer::new();

pub fn with_config(config: ActivationConfig) -> Self

Create a new activation visualizer with custom configuration

pub fn register( &mut self, layer_name: &str, values: Vec<f32>, shape: Vec<usize>, ) -> Result<()>

Register activations from a layer

Arguments

• layer_name - Name of the layer
• values - Flattened activation values
• shape - Original shape of the activation tensor

Example

let activations = vec![0.1, 0.5, 0.3, 0.8];
visualizer.register("layer1", activations, vec![2, 2]).unwrap();

pub fn get_activations(&self, layer_name: &str) -> Option<&ActivationData>

Get activations for a specific layer

pub fn get_layer_names(&self) -> Vec<String>

Get all layer names with registered activations

pub fn create_histogram(&self, layer_name: &str) -> Result<ActivationHistogram>

Create a histogram of activation values

pub fn create_heatmap( &self, layer_name: &str, reshape: Option<(usize, usize)>, ) -> Result<ActivationHeatmap>

Create a heatmap from 2D activations

Arguments

• layer_name - Name of the layer
• reshape - Optional reshape dimensions (e.g., [height, width])

pub fn export_statistics( &self, layer_name: &str, output_path: &Path, ) -> Result<()>

Export activation statistics to JSON

pub fn plot_distribution_ascii(&self, layer_name: &str) -> Result<String>

Plot distribution as ASCII histogram

pub fn print_summary(&self) -> Result<String>

Print summary statistics for all layers

pub fn clear(&mut self)

Clear all stored activations

pub fn num_layers(&self) -> usize

Get number of stored activations

Trait Implementations

impl Debug for ActivationVisualizer

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ActivationVisualizer

fn default() -> Self

Returns the “default value” for a type.