osc.viz.attentions

Attention visualization utils: single attn maps and rollouts.

Functions

viz_slot_attns
viz_slot_rollout_all_options	Visualize slot attention rollout: per-iteration or final normalization.
viz_vit_attns
viz_vit_rollout_all_options	Visualize backbone rollout: residual adjustment, head reduction