← Walkthroughsrc/speech_decoding/models/v14_converged_v3/pack_r4.py
Selected training excerpt · lines 241–243 · source SHA-256 10e2eebf5adf
241 out = fill.to(h_vis.dtype).expand(B, total, d).clone()242 out.scatter_(1, idx[:, :, None].expand(B, m_vis, d), h_vis)243 return out