抱歉標(biāo)題措辭不佳。我想做的是這樣的： 矩陣 1 是原始矩陣，矩陣 2 是矩陣 1，但每個(gè)偶數(shù)列和行都被取出。矩陣 3 是矩陣 1，但只有 1 (mod 3) 列和行。矩陣 4 相同，具有 1 (mod 4) 列和行。矩陣 5 有 1 (mod 2) 列和所有行。是否有以這種方式快速且可以利用 GPU 來(lái)操縱張量的 PyTorch 函數(shù)？這有點(diǎn)像 MaxPool2d，但我只需要第一個(gè)值而不是最大值。如果沒(méi)有這樣的功能，有沒(méi)有辦法手動(dòng)完成但速度仍然很快？
1 回答
繁星點(diǎn)點(diǎn)滴滴
TA貢獻(xiàn)1803條經(jīng)驗(yàn) 獲得超3個(gè)贊
Matrix 5 是最容易展示的，因?yàn)槟阒恍枰刂粋€(gè)維度切片。但是您可以將兩者切分以獲得其他結(jié)果。
matrix5 = matrix1[, ::2]
這種表示法每隔一列取一次，從第 0 列開(kāi)始。
添加回答
舉報(bào)
0/150
提交
取消