osc.viz.embeds

Visualization of positional embeddings.

Functions

viz_positional_embedding

Visualize 2D positional embedding, possibly resized to a different grid size.