p-doom/crowd-code-dataset-0.2 huggingface dataset:
[content] models/dynamics.py (python)
        # Pad action embeddings to match video shape: (B, T-1, 1, M) -> (B, T, 1, M)
        act_embed_padded = jnp.pad(act_embed_BTm11M, ((0, 0), (1, 0), (0, 0), (0, 0)))
        # Broadcast to match video shape: (B, T, 1, M) -> (B, T, N, M)
        act_embed_broadcast = jnp.broadcast_to(act_embed_padded, vid_embed_BTNM.shape)
        vid_embed_BTNM += act_embed_broadcast
        logits = self.transformer(vid_embed_BTNM)