H,W,C),分割(此時(shí)用reshape)應(yīng)該先按照(B,H/H分割size,H分割size,W/W分割size,W分割size,C)展開(kāi),然后再用permute,或者tanspose轉(zhuǎn)化為(B,H/H分割size,W/W分割size,H分割size,W分割size,C),最后,再用contiguous().view轉(zhuǎn)化為(B,(H/H分割size)*(W/W分割size),H分割size,W分割size,...
www.dbjr.com.cn/python/328694h...htm 2025-5-26