1. Pytorch中的廣播機(jī)制 如果一個(gè)Pytorch運(yùn)算支持廣播的話,那么就意味著傳給這個(gè)運(yùn)算的參數(shù)會(huì)被自動(dòng)擴(kuò)張成相同的size,在不復(fù)制數(shù)據(jù)的情況下就能進(jìn)行運(yùn)算,整個(gè)過(guò)程可以做到避免無(wú)用的復(fù)制,達(dá)到更高效的運(yùn)算。 廣播機(jī)制實(shí)際上是在運(yùn)算過(guò)程中,去處理兩個(gè)形狀不同向量的一種手段。
www.dbjr.com.cn/article/2714...htm 2025-5-30