這篇文章主要介紹了pytorch torch.expand和torch.repeat的區(qū)別詳解,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧 1.torch.expand 函數(shù)返回張量在某一個維度擴(kuò)展之后的張量,就是將張量廣播到新形狀。函數(shù)對返回的張量不會分配新內(nèi)存,即在原始張量上
www.dbjr.com.cn/article/1735...htm 2025-5-19