遼寧響應(yīng)式網(wǎng)站費(fèi)用北京網(wǎng)絡(luò)推廣優(yōu)化公司
broadcast:
1、能夠進(jìn)行維度擴(kuò)展,是自動(dòng)的2、在擴(kuò)展的時(shí)候不需要拷貝數(shù)據(jù)
要點(diǎn):
- 從最小的維度開始匹配,如果前面沒有維度了,在前面插入一個(gè)新的維度。- 插入的新維度size是1,再將其擴(kuò)展為與目標(biāo)相同大小size。
一個(gè)實(shí)際案例:
[class, students, scores]
[4, 32, 8 ]
# 有4個(gè)班,每個(gè)班32個(gè)學(xué)生,每個(gè)學(xué)生都有8門課程由于今年期末考試難度過大,校長(zhǎng)決定給每個(gè)同學(xué)各科成績(jī)加5分
5.0 -> [1]
如果手動(dòng)操作:[1].unsqueeze(0).unsqueeze(0).expand(4, 32, 8)
# 一個(gè)浮點(diǎn)數(shù)占用4byte,如果擴(kuò)張成[4,32,8]的shape需要占用4096byte,broadcast能大大節(jié)約內(nèi)存
注:在最小維度匹配,才能繼續(xù)擴(kuò)展