求大神解读一下这行代码的含义

#30

by bluelueSea - opened Sep 27

Sep 27

kv_mem = (last_recurrent_state * k_t.unsqueeze(-1)).sum(dim=-2)

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment