RWKV的输入是B L C patch是想要将原来的分成4*4块大小 这怎么做？AI给出的是 k_local = rearrange(k, 'b (h w) c -> b c h w', h=H, w=W)
v_local = rearrange(v, 'b (h w) c -> b c h w', h=H, w=W)
ph, pw = 4, 4 # 定义块的高度和宽度
k_local = rearrange(k_local, 'b c (nh ph) (nw pw) -> b (nh nw) (ph pw) c', ph=ph, pw=pw)
v_local = rearrange(v_local, 'b c (nh ph) (nw pw) -> b (nh nw) (ph pw) c', ph=ph, pw=pw) 这完全不匹配啊
请问您这个是哪篇论文提及的方法？他的这个方法我感觉似乎有点奇怪