osc.viz.rollout

Functions

cross_attn_rollout	Cross attention rollout: how much an object token attends to context tokens.
self_attn_rollout	Self-attn rollout: how much output token(s) attend to input tokens across layers
slot_attn_rollout	Slot attention rollout: how much a slot token attends to context tokens.